因此,如果我有以下数组arr:
>>> arr
array([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14],
[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14]])
现在,如果我想获得第一行,我会这样做:
>>> arr[0]
array([0, 1, 2, 3, 4])
但是,当我使用np.where来定位特定行时,例如:
>>> np.where(arr == [0,1,2,3,4])
我得到这个输出
(array([0, 0, 0, 0, 0, 3, 3, 3, 3, 3], dtype=int64),
array([0, 1, 2, 3, 4, 0, 1, 2, 3, 4], dtype=int64))
然而,这不是我想要的。我想改为获取行索引。e、 g:
(array([0, 3], dtype=int64)
有没有办法做到这一点?非常感谢您的任何建议
我想您应该检查这些行是否等于给定的数组。在这种情况下,您需要
all
:相关问题 更多 >
编程相关推荐