我需要以某种方式制作该数组:
[[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]]])