带有numpy数组的布尔表达式[重复]

Question

基本上我正在尝试使用 numpy 数组的布尔表达式，例如：

import numpy as np
a = np.array([1,0,1])
b = np.array([0,1,1])
c = np.array([1,0,1])

if (a ==b) or (a==c):
    d = [2,5,5]
else : d = [1,5,5]
print d

这会返回一个错误

ValueError：具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()

编写此代码的正确方法是什么？

Answer 1

条件应为：

if (a==b).all() or (a==c).all():

看到这个question