Construire un tableau 3d en numpy à partir d'un tableau 2d existant

Question

Pendant la préparation des données pour NumPy calculer. Je suis curieux de façon à construire:

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

de:

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

J'essaie d'utiliser la commande NumPy:

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

mais il semble fonctionne pas!

Answer 1

hstack et vstack ne pas modifier le nombre de dimensions des tableaux: ils simplement les mettre « côte à côte ». Ainsi, la combinaison de tableaux bidimensionnels crée un nouveau tableau bidimensionnel (pas un tableau 3D!).

Vous pouvez faire ce que Daniel a suggéré (utiliser directement numpy.array([d1, d2])).

Vous pouvez alternativement convertir vos tableaux de tableaux 3D avant de les empiler, en ajoutant une nouvelle dimension à chaque tableau:

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

En fait, d1[newaxis,...].shape == (1, 18, 18), et vous pouvez empiler les deux tableaux 3D directement et obtenir le nouveau Tableau 3D (d3) que vous vouliez.

Answer 2

Il suffit de faire d3 = array([d1,d2]) semble fonctionner pour moi:

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