2025年6月4日 星期三 乙巳(蛇)年 三月初八 设为首页 加入收藏
rss
您当前的位置:首页 > 计算机 > 编程开发 > Python

Python Collection 小技巧

时间:12-14来源:作者:点击数:4

判断一个 list 是否为空

传统的方式:

  • 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

遍历 list 的同时获取索引

传统的方式:

  • 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

list 排序

在包含某元素的列表中依据某个属性排序是一个很常见的操作。例如这里我们先创建一个包含 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 方法,作用如其字面含义。

在 Dictionary 中元素分组

和上面类似,先创建 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 类型的值。

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门
本栏推荐