Конструировать 3d-массив в numpy из существующего 2d-массива

Question

Во время подготовки данных для вычисления NumPy. Мне любопытно, как построить:

myarray.shape => (2,18,18) 

от:

d1.shape => (18,18) 
d2.shape => (18,18) 

Я пытаюсь использовать команду NumPy:

hstack([[d1],[d2]]) 

, но это выглядит не работает!

Answer 1

hstack и vstack не делать никаких изменений количества размерностей массивов: они просто положить их «бок о бок». Таким образом, объединение двумерных массивов создает новый двумерный массив (а не трехмерный)!

Вы можете сделать то, что предложил Даниил (напрямую использовать numpy.array([d1, d2])).

Вы можете альтернативно преобразовать массивы в 3D массивы перед укладкой их, добавляя новое измерение в каждом массиве:

d3 = numpy.vstack([ d1[newaxis,...], d2[newaxis,...] ]) # shape = (2, 18, 18) 

В самом деле, d1[newaxis,...].shape == (1, 18, 18), и вы можете сложить обе 3D массивы непосредственно и получить новый 3D-массив (d3), который вы хотели.

Answer 2

Просто делать d3 = array([d1,d2]), кажется, работает для меня:

>>> from numpy import array 
>>> # ... create d1 and d2 ... 
>>> d1.shape 
(18,18) 
>>> d2.shape 
(18,18) 
>>> d3 = array([d1, d2]) 
>>> d3.shape 
(2, 18, 18)