Contruir matriz 3D en numpy desde la matriz 2d existente

Question

Durante la preparación de los datos para el cálculo de NumPy. Tengo curiosidad por forma de construir:

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

de:

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

intento utilizar comandos NumPy:

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

pero no parece que va trabajar!

Answer 1

hstack y vstack hacer ningún cambio en el número de dimensiones de las matrices: se limitan a los pusieron "lado a lado". Por lo tanto, la combinación de matrices bidimensionales crea una nueva matriz bidimensional (¡no una 3D!).

Puede hacer lo que Daniel sugirió (use directamente numpy.array([d1, d2])).

Puede convertir alternativamente las matrices a las matrices en 3D antes de apilarlas, añadiendo una nueva dimensión a cada matriz:

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

De hecho, d1[newaxis,...].shape == (1, 18, 18), y se puede apilar ambas matrices 3D directamente y obtener el nuevo Matriz 3D (d3) que usted quería.

Answer 2

Sólo haciendo d3 = array([d1,d2]) parece que funciona para mí:

>>> 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)