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

Con*_*ezy 3 python numpy

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

array1 = np.array([1,1,1,1,1]) would return True
array2 = np.array([1,0,1,0,1]) would return False
Run Code Online (Sandbox Code Playgroud)

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

moz*_*way 5

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

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

>>> len(set(array2)) == 1
False
Run Code Online (Sandbox Code Playgroud)