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

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

Answer 2

这也有效，并且似乎比其他一些方法略快：

>>> array1 = np.array([1,1,1,1,1])
>>> array2 = np.array([1,0,1,0,1])
>>> (array1 == array1[0]).all()
True
>>> (array2 == array2[0]).all()
False

与其他解决方案相比，10k 阵列的一些非常粗略的时序图：

• 这种方法：1.07 毫秒
• np.isin() ：1.23 毫秒
• set() ：2.63 毫秒

Answer 3

这可以使用- np.unique

import numpy as np

array1 = np.array([1,1,1,1,1])
#following passes assertion test
assert(len(np.unique(array1, return_counts=True)[0])==1)

#without getting counts of unique values
#assert(len(np.unique(array1))==1)

array2 = np.array([1,0,1,0,1])
#following will throw an assertion error
assert(len(np.unique(array2, return_counts=True)[0])==1)

Answer 4

我的尝试是将所有项目与第一个项目进行比较，然后检查结果是否包含False ：

import numpy as np
array1 = np.array([1,1,1,1,1])
print(not np.isin(False, array1 == array1[0]))

Answer 5

您可以尝试使用all() ：

例如：

all(array1)
  > True

all(array2)
  > False