Python numpy数组操作

我需要以某种方式制作该数组：

[[639 190]
 [ 44   1]
 [ 71   4]
 ...,[863 347]
 [870 362]
 [831 359]]

看起来像这样：

[[[639 190]]
 [[ 44   1]]
 [[ 71   4]]
 ...,[[863 347]]
 [[870 362]]
 [[831 359]]]

我该怎么办？我刚接触numpy,我需要它用于我的科学实验.

解决方法

用 None/np.newaxis添加一个新轴 –

a[:,None,:] # Or simply a[:,None]

样品运行 –

In [222]: a = np.random.randint(0,9,(4,3))

In [223]: a
Out[223]: 
array([[1,6,6],[4,4,5],[7,4],1,3]])

In [224]: a[:,None]
Out[224]: 
array([[[1,6]],[[4,5]],[[7,4]],3]]])