python - 如何检查数组的所有值是否彼此相等？

Question

我想检查一个数组中是否包含所有相同的值。示例如下。

array1 = np.array([1,1,1,1,1]) would return True
array2 = np.array([1,0,1,0,1]) would return False

我知道如何检查数组中的所有值是否都等于某个值。但我想检查数组中的所有值是否彼此相等，无论值是什么。有没有办法在不创建函数的情况下只使用 Numpy 来做到这一点？

Answer 1

您可以使用 python 集。如果集合的长度为1，则所有值都相同:

>>> len(set(array1)) == 1
True

>>> len(set(array2)) == 1
False