Python俱乐部
Python
小课题
京东优惠券
独立软件开发者 Alex Marandon 在他的博客中介绍了数个关于 Python Collection 的实用小技巧,在此与诸位分享。
传统的方式:
if len(mylist): # Do something with my list else: # The list is empty
由于一个空 list 本身等同于 False,所以可以直接:
if mylist: # Do something with my list else: # The list is empty
传统的方式:
i = 0 for element in mylist: # Do something with i and element i += 1
这样更简洁些:
for i, element in enumerate(mylist): # Do something with i and element pass
在包含某元素的列表中依据某个属性排序是一个很常见的操作。例如这里我们先创建一个包含 person 的 list:
class Person(object): def __init__(self, age): self.age = age persons = [Person(age) for age in (14, 78, 42)]
传统的方式是:
def get_sort_key(element): return element.age for element in sorted(persons, key=get_sort_key): print "Age:", element.age
更加简洁、可读性更好的方法是使用 Python 标准库中的 operator 模块:
from operator import attrgetter for element in sorted(persons, key=attrgetter('age')): print "Age:", element.age
attrgetter 方法优先返回读取的属性值作为参数传递给 sorted 方法。operator 模块还包括 itemgetter 和 methodcaller 方法,作用如其字面含义。
和上面类似,先创建 Persons:
class Person(object): def __init__(self, age): self.age = age persons = [Person(age) for age in (78, 14, 78, 42, 14)]
如果现在我们要按照年龄分组的话,一种方法是使用 in 操作符:
persons_by_age = {} for person in persons: age = person.age if age in persons_by_age: persons_by_age[age].append(person) else: persons_by_age[age] = [person] assert len(persons_by_age[78]) == 2
相比较之下,使用 collections 模块中 defaultdict 方法的途径可读性更好:
from collections import defaultdict persons_by_age = defaultdict(list) for person in persons: persons_by_age[person.age].append(person)
defaultdict 将会利用接受的参数为每个不存在的 key 创建对应的值,这里我们传递的是 list,所以它将为每个 key 创建一个 list 类型的值。