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.all()，它接受​​一个布尔数组并检查所有True，并进行相等比较：

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

Answer 2

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

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