统计分组元素
根据给定的函数对列表的元素进行分组,并返回每个分组中元素的数量。
- 使用
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}