带有 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

这returns一个错误

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

正确的编码方式是什么？

Answer 1

条件应为：

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

看到这个question

带有 numpy 数组的布尔表达式

Boolean expression with numpy arrays

python

arrays

numpy

boolean-expression