[파이썬 numpy] 비어있는 배열 만들고 원소 추가하기

비어있는 배열을 만들고 점(point)와 같은 원소를 추가해야 하는 상황에서 사용하는 방법입니다. 비어있는 배열에 아래와 같이 3차원 점들을 for 문을 이용하여 하나씩 추가해주려고 합니다.

points = np.array([[4, 5, 6],
                   [7, 8, 9],
                   [10, 11, 12],
                   [10, 11, 12],
                   [10, 11, 12]])

 

비어있는 배열을 하나 만들어줍니다. 추가할 배열들의 차원과 일치하는 크기로 만들어주어야 합니다.

ar=np.empty([0,3])

 

위 코드를 잘 이해해야 합니다. 행렬로 생각해보면 0행 3열입니다. x,y,z 라는 열 이름만 만들어 놓은 상태라고 생각하시면 됩니다.

np.vstack 을 이용하여 점들을 원소로 추가해줍니다. 행렬에서 행을 추가해주는 것으로 이해하시면 됩니다. 각 행이 점입니다.

for point in points :
    ar=np.vstack([ar,np.dot(T,point)])