我一直在发疯,试图找出我在这里做错了什么愚蠢的事情。
我正在使用NumPy,并且我想从中选择特定的行索引和特定的列索引。这是我的问题的要点:
import numpy as np
a = np.arange(20).reshape((5,4))
print a[[0, 1, 3], :]
print a[[0, 1, 3], 2]
print a[[0,1,3], [0,2]]
为什么会这样呢?我当然应该能够选择第一行,第二行和第四行以及第一列和第三列?我期望的结果是:
a[[0,1,3], [0,2]] => [[0, 2], [4, 6], [12, 14]] 问题来源于stack overflow
花式索引要求您提供每个维度的所有索引。您为第一个提供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]])
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。