J'ai un 2 dimensions: A = numpy.array([[1, 2, 3], [4, 5, 6]]) et voudrais le convertir en un 3 dimensions tableau: B = numpy.array([[[1, 2, 3], [4, 5, 6]]])Numpy: convertir tableau 2D tableau 3D
est-il un moyen simple de le faire?
ajouter simplement un nouvel axe au début avec np.newaxis -
import numpy as np
B = A[np.newaxis,:,:]
Nous pourrions ignorer la liste des axes de fuite -
B = A[np.newaxis]
Aussi, apportez l'alias None pour remplacer np.newaxis pour un plus solution compacte -
B = A[None]
Il est également possible de créer un nouveau tableau NumPy en utilisant le constructeur afin qu'il prenne une liste. Cette liste contient un seul élément qui est le tableau A et il vous permettra de créer le même tableau avec la dimension singleton étant le premier. Le résultat serait le tableau 3D que vous désirez:
B = numpy.array([A])
In [13]: import numpy as np
In [14]: A = np.array([[1, 2, 3], [4, 5, 6]])
In [15]: B = np.array([A])
In [16]: B
Out[16]:
array([[[1, 2, 3],
[4, 5, 6]]])