[Numpy] numpy 배열 합치기

numpy 배열들을 합치는 다양한 방법들을 알아보자.

---

vstack 함수, hstack 함수

합치는 array를 하나의 iterable 객체에 담아 인자로 넘겨야 된다.

np.vstack((array1, array2)) : vertical 방향으로 쌓음

np.hstack((array1, array2)) : horizontal 방향으로 쌓음

a = np.array([1,2,3])
b = np.array([4,5,6])

print(np.vstack((a, b)))
# 출력:
# [[1 2 3]
#  [4 5 6]]

print(np.hstack([a, b]))
# 출력: [1 2 3 4 5 6]

 

concatenate 함수

axis를 활용해 numpy 배열을 합치는 방향을 결정한다.

a = np.array([[1,2,3]]) # array가 1차원 array라면 axis=1을 적용할 수 없다.
b = np.array([[4,5,6]])

print(np.concatenate((a, b), axis = 0)) # 열 방향으로 쌓음
# 출력:
# [[1 2 3]
#  [4 5 6]]

print(np.concatenate((a, b), axis = 1)) # 행 방향으로 쌓음
# 출력: [[1 2 3 4 5 6]]