[Python] Array 합치기

두 개의 Numpy Array를 합치는 방법에 대해서 배워보도록 하겠습니다.

 

Numpy 배열을 합치는 방법으로 크게 두 가지 함수를 사용하는 방법이 있습니다.

 

arr1 = np.array([[1, 2],
                 [3, 4]])
arr2 = np.array([[5, 6], 
                 [7, 8]])

 

합칠 2개의 Array를 만들어 보았습니다.

 

1. np.concatenate

 

arr3 = np.concatenate([arr1, arr2], axis = 0)
print(arr3.shape)
arr3

(4, 2)
array([[1, 2],
       [3, 4],
       [5, 6],
       [7, 8]])
       
       
       
arr3 = np.concatenate([arr1, arr2], axis = 1)
print(arr3.shape)
arr3

(2, 4)
array([[1, 2, 5, 6],
       [3, 4, 7, 8]])

 

axis 인자에 따라 배열을 합치는 기준이 달라집니다.

axis = 0 일 경우 0번째 축에 대해서 합쳐집니다. 

 

 

2. np.stack

 

arr4 = np.stack([arr1,arr2])
print(arr4.shape)
arr4

(2, 2, 2)
array([[[1, 2],
        [3, 4]],

       [[5, 6],
        [7, 8]]])

 

np.stack의 경우 새로운 차원으로 2개의 Array가 합쳐집니다.