如何制作一个3D数组的二维numpy数组？

Question

我有一个二维数组与形状（x，y），我想要转换为形状（x，y，1）的3d数组。有没有一个很好的Pythonic方法来做到这一点？

Answer 1

除了其他的答案，你也可以使用与numpy.newaxis切片：

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

甚至这个（这将尺寸任意数量的工作）：

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