写点什么

Python 代码阅读(第 7 篇):列表分组计数

用户头像
Félix
关注
发布于: 19 分钟前
Python代码阅读(第7篇):列表分组计数

本篇阅读的代码实现了使用分组函数对列表进行分组,并计算每组的元素个数的功能。


本篇阅读的代码片段来自于30-seconds-of-python

count_by

def count_by(arr, fn=lambda x: x):  key = {}  for el in map(fn, arr):    key[el] = 1 if el not in key else key[el] + 1  return key
# EXAMPLESfrom math import floorcount_by([6.1, 4.2, 6.3], floor) # {6: 2, 4: 1}count_by(['one', 'two', 'three'], len) # {3: 2, 5: 1}
复制代码


count_by根据给定的函数对列表中的元素进行分组,并返回每组中元素的数量。该使用 map()使用给定函数映射给定列表的值。在映射上迭代,并在每次出现时增加元素数。


该函数使用not in判断目前字典中是否含有指定的key,如果不含有,就将该key加入字典,并将对应的value设置为1;如果含有,就将value1

使用字典推导式

在** Python 代码阅读:根据给定的函数对列表中的元素进行分组**中使用了字典推导式,将列表进行了分组。这里也可以使用同样的方式,在分组之后直接获取列表长度。不过这种写法遍历了两次列表,会使程序效率变低。


def count_by(lst, fn):  return {key : len([el for el in lst if fn(el) == key]) for key in map(fn, lst)}
复制代码

使用collections.defaultdict简化代码

class collections.defaultdict([default_factory[, ...]])
复制代码


collections.defaultdict包含一个default_factory属性,可以用来快速构造指定样式的字典。


当使用int作为default_factory,可以使defaultdict用于计数。因此可以直接使用它来简化代码。相比字典推导式的方法,只需要对列表进行一次循环即可。


 from collections import defaultdict
def count_by(lst, fn): d = defaultdict(int) for el in lst: d[fn(el)] += 1 return d
复制代码


当使用 list 作为 default_factory时,很轻松地将(键-值对组成的)序列转换为(键-列表组成的)字典。因此我们也可以据此改写** Python 代码阅读:根据给定的函数对列表中的元素进行分组**中的实现方式,提高效率。


def group_by(lst, fn):  d = defaultdict(list)  for el in lst:    d[fn(el)].append(el)  return d
# EXAMPLESfrom math import floorgroup_by([6.1, 4.2, 6.3], floor) # {4: [4.2], 6: [6.1, 6.3]}group_by(['one', 'two', 'three'], len) # {3: ['one', 'two'], 5: ['three']}
复制代码


发布于: 19 分钟前阅读数: 2
用户头像

Félix

关注

还未添加个人签名 2018.05.04 加入

还未添加个人简介

评论

发布
暂无评论
Python代码阅读(第7篇):列表分组计数