python - frozenset 至少 x 个元素

Question

我目前有这段代码，它检查数组中的所有元素是否相同。如果是这种情况，则返回 true

def all_equal(lst):
  """
  >>> all_equal([1,1,1,1,1,1,1])
  True
  >>> all_equal([1,2,3,1])
  False
  """
  return len(frozenset(lst)) == 1

但我想检查的是是否至少有 5 个相同的元素。

这样

[1,1,1,1,1,2,2]

也将返回 True。因为有 5 次 1

Answer 1

使用collections.Counter() :

from collections import Counter

def all_equal(lst, count):
    return any(v >= count for v in Counter(lst).values())