统计分组元素

根据给定的函数对列表的元素进行分组,并返回每个分组中元素的数量。

  • 使用collections.defaultdict初始化一个字典。
  • 使用map()函数使用给定的函数对给定列表的值进行映射。
  • 遍历映射结果,并每次出现时增加元素计数。
from collections import defaultdict
from math import floor

def count_by(lst, fn = lambda x: x):
  count = defaultdict(int)
  for val in map(fn, lst):
    count[val] += 1
  return dict(count)

count_by([6.1, 4.2, 6.3], floor) # {6: 2, 4: 1}
count_by(['one', 'two', 'three'], len) # {3: 2, 5: 1}