Numpy：如何使用1d数组索引2d数组_CDA答疑社区

热线电话：13121318867

登录

啊啊啊啊啊吖

2019-03-07 阅读量: 774

Numpy：如何使用1d数组索引2d数组

我有一个2d数组：

a = np.random.randint(100, size=(6, 4))

[[72 76 40 11]

[48 82 6 87]

[53 24 25 99]

[ 7 94 82 90]

[28 81 10 9]

[94 99 67 58]]

和1d数组：

idx = np.random.randint(4, size=6)

[0, 3, 2, 1, 0, 2]

是否可以索引2d数组以便得到结果

a[idx]

[72, 87, 25, 94, 28, 67]

解决办法：由于您具有列索引，因此您只需要行索引。你可以生成那些arange。

>>> a[np.arange(len(a)), idx]

array([72, 87, 25, 94, 28, 67])

6.4876

1

2

关注作者

收藏

评论(0)

发表评论

暂无数据

CDA考试动态

CDA报考指南

推荐帖子