Contrasta l'array 3d in numpy dall'array 2d esistente

Question

Durante la preparazione dei dati per il calcolo NumPy. Sono curioso di modo di costruire:

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

da:

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

cerco di usare il comando NumPy:

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

ma sembra non funziona!

Answer 1

hstack e vstack non fanno cambiare il numero delle dimensioni delle matrici: non fanno che li hanno messi "fianco a fianco". Pertanto, la combinazione di array bidimensionali crea un nuovo array bidimensionale (non uno 3D!).

È possibile eseguire le operazioni suggerite da Daniel (utilizzare direttamente numpy.array([d1, d2])).

È possibile, in alternativa convertire le matrici a matrici 3D prima di accatastamento, con l'aggiunta di una nuova dimensione per ogni matrice:

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

Infatti, d1[newaxis,...].shape == (1, 18, 18), ed è possibile impilare entrambi gli array 3D direttamente e ottenere il nuovo 3D array (d3) che volevi.

Answer 2

Solo facendo d3 = array([d1,d2]) sembra funzionare per me:

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