我需要以某种方式制作该数组:
[[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, 4],
[4, 1, 3]])
In [224]: a[:,None]
Out[224]:
array([[[1, 6, 6]],
[[4, 4, 5]],
[[7, 4, 4]],
[[4, 1, 3]]])