栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

从NumPy数组中选择特定的行和列

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

从NumPy数组中选择特定的行和列

花式索引要求您提供每个维度的所有索引。您为第一个提供3个索引,为第二个仅提供2个索引,因此会出现错误。您想做这样的事情:

>>> a[[[0, 0], [1, 1], [3, 3]], [[0,2], [0,2], [0, 2]]]array([[ 0,  2],       [ 4,  6],       [12, 14]])

当然写这很痛苦,所以您可以让广播帮助您:

>>> a[[[0], [1], [3]], [0, 2]]array([[ 0,  2],       [ 4,  6],       [12, 14]])

如果您使用数组而不是列表建立索引,则此操作要简单得多:

>>> row_idx = np.array([0, 1, 3])>>> col_idx = np.array([0, 2])>>> a[row_idx[:, None], col_idx]array([[ 0,  2],       [ 4,  6],       [12, 14]])


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/637660.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号