[파이썬] 행렬에 행 추가 & 행 합치기 하는 방법(numpy.vstack)

파이썬에서 단순 array라면, append를 통해

A = []
A.append([1, 2, 3])
A.append([4, 5, 6])
A = np.array(A)
print(A)

[[1 2 3]

[4 5 6]]

라는 결과를 쉽게 얻을 수 있겠지만. numpy 라이브러리에서는 행렬에 append가 적용되지 않습니다.

 

하지만, numpy 라이브러리의 행렬에서도 numpy.vstack를 통해 행을 추가할 수 있습니다.

import numpy as np

A = np.array([[1,2,3,4],[5,6,7,8]])
B = np.array([[9,10,11,12],[13,14,15,16]])
C = np.vstack([A, B])

print(A)
print(B)
print('\n')
print(C)

A는 [[1, 2, 3, 4], [5, 6, 7, 8]], B는 [[9, 10, 11, 12], [13, 14, 15, 16]]에서 np.vstack를 적용하면

 

[[ 1 2 3 4]

[ 5 6 7 8]

[ 9 10 11 12]

[13 14 15 16]]

 

라는 결과값이 나오게 됩니다.

 

단, 열의 크기가 같아야 추가 가능하니 유의바랍니다.