Python numpy数组操作

发布时间：2020-12-20 12:01:35 所属栏目：Python 来源：网络整理

导读：我需要以某种方式制作该数组： [[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,我需要它用于我的科学实验. 解决方

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

[[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]]])

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!