두 개의 2차원 넘파이 배열에 0축(axis=0)을 추가하여 하나로 합치기

두 개의 2차원 넘파이 배열에 0축(axis=0)을 추가하여 하나로 합치는 예제 코드입니다.


2021. 9. 30 - 최초 작성


예를 들어 2차원 배열의 shape가 (10,3)인 경우 2개를 합치게 되면 (2, 10, 3)이 됩니다. 

(10, 3) + (10, 3) -> (2, 10, 3)


전체 소스코드입니다.

 

import numpy as np


a = np.arange(30).reshape(10,3)
b = np.arange(30).reshape(10,3)


print(a.shape, b.shape)


a = np.expand_dims(a, axis=0)
b = np.expand_dims(b, axis=0)

print(a.shape, b.shape)


c = np.concatenate((a, b), axis=0)

print(c.shape)

 

 

실행 결과입니다.

 

(10, 3) (10, 3)
(1, 10, 3) (1, 10, 3)
(2, 10, 3)