Numpy:我如何确定numpy数组的所有元素是否等于数字

Question

我需要知道数组的所有元素numpy是否等于数字

这将是:

numbers = np.zeros(5) # array[0,0,0,0,0]
print numbers.allEqual(0) # return True because all elements are 0

我可以制作一个算法但是,在numpy库中实现了一些方法吗？

Answer 1

如果您想比较浮点数，请np.isclose改用：

np.all(np.isclose(numbers, numbers[0]))

Answer 2

你可以把它分解成np.all()一个布尔数组并检查它的全部True,并进行相等比较:

np.all(numbers == 0)
# or equivalently
(numbers == 0).all()