행렬 및 리스트 값 0으로 바꿀 때의 방법 비교하기

리스트를 이용한 행렬 구현에서 조건에 해당하는 요소를 특정 값 x로 바꾸려고 할 때와

넘파이를 이용한 행렬 구현에서 조건에 해당하는 요소를 특정 값 x로 바꾸는 방법에 대해서 비교하며

알아보자.

 

조건은 0보다 작은 음수에 대해서 값을 0으로 바꾼다.

 

1. 리스트 이용

특정 값 = 0, 2차원 정사각행렬

# 리스트를 이용한 행렬 구현
x = [[1,2],[-3,-4]]

# 모든 요소 접근을 위한 중첩 for문 사용, 여기서는 특정 값을 0으로 설정
for i in range(len(x)):
    for j in range(len(x)):
        if x[i][j]<0:
            x[i][j] = 0

 

 

2. 넘파이 이용

import numpy as np
x = np.array([[1,2],[-3,-4]])
mask = (x<=0)
print(mask)

x[mask] = 0
print(x)

x<=0을 통해서 해당되는 조건의 값에는 True / False를 반환하기에 이를 이용하면 쉽게 구현할 수가 있다.