Прикрепите два массива 3d numpy в Python

Question

Мне было интересно, как я могу прикрепить два массива 3d numpy в python?

Например, у меня есть один с формой (81,81,61), и я хотел бы получить вместо этого массив формы (81,81,122), присоединив исходный массив к себе в направлении z.

Answer 1

Один из способов заключается в использовании np.dstack, который сцепляет массивы вдоль третьей оси (г для глубины).

Например:

>>> a = np.arange(8).reshape(2,2,2) 
>>> np.dstack((a, a)) 
array([[[0, 1, 0, 1], 
     [2, 3, 2, 3]], 

     [[4, 5, 4, 5], 
     [6, 7, 6, 7]]]) 

>>> np.dstack((a, a)).shape 
(2, 2, 4) 

Вы также можете использовать np.concatenate((a, a), axis=2).