python - 返回两个numpy数组中的哪些元素都等于X值？

Question

例如，A = [5, 0, 0, 5] B = [5, 5, 0, 5]

我想获取两个数组中哪些元素等于 5

输出应该是 =[0, 3]

标签： pythonnumpy

Answer 1

np.where(np.logical_and(A == 5, B == 5))