Jak uczynić tablicę 2d numpy tablicą 3d?

Question

Mam tablicę 2d o kształcie (x, y), którą chcę przekonwertować na tablicę 3d o kształcie (x, y, 1). Czy istnieje przyjemny sposób Pythonic to zrobić?

Answer 1

W uzupełnieniu do innych odpowiedzi, można użyć również krojenie z numpy.newaxis:

>>> from numpy import zeros, newaxis 
>>> a = zeros((6, 8)) 
>>> a.shape 
(6, 8) 
>>> b = a[:, :, newaxis] 
>>> b.shape 
(6, 8, 1) 

lub nawet ten (który będzie współpracować z dowolnej liczbie wymiarów):

>>> b = a[..., newaxis] 
>>> b.shape 
(6, 8, 1) 

Answer 2

import numpy as np 

a= np.eye(3) 
print a.shape 
b = a.reshape(3,3,1) 
print b.shape 

Answer 3

numpy.reshape(array, array.shape + (1,))