python - Numpy.where 解决方法

Question

对于我的数组:
数组([[ 1, 2, 3, 4, 5], #a
[ 1, 3, 5, 7, 9],
[5, 10, 15, 20, 25],
[2, 4, 6, 8, 5]])

和
数组([[ 1, 2, 3, 4, 16], #b
[ 1, 3, 16, 7, 9],
[ 5, 16, 15, 20, 25],
[2, 4, 6, 8, 5]])

我尝试获取 np.where((a==5 and b==16)) 的结果。我希望:

输出[1]: (array([0, 1], dtype=int64), array([4, 2], dtype=int64),

因为那是 5 和 16 共享相同索引的地方；但是我得到了一个

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

当我尝试 np.where((a.all()==5 and b.any()==16)) 时，我得到了

输出[1]: (array([], dtype=int64),)

有什么想法吗？提前致谢。

Answer 1

您想使用 & 而不是 and:

np.where((a==5) & (b==16))

在处理 numpy 数组时，您希望使用按位运算符而不是逻辑与。