Python 俱乐部

本页面只读。您可以查看源文件，但不能更改它。如果您觉得这是系统错误，请联系管理员。
====== Python Collection 小技巧 ======

独立软件开发者 Alex Marandon 在他的博客中介绍了数个关于 Python Collection 的实用小技巧，在此与诸位分享。

===== 判断一个 list 是否为空 =====


传统的方式：
<code python>
if len(mylist):
    # Do something with my list
else:
    # The list is empty
</code>
由于一个空 list 本身等同于 False，所以可以直接：
<code python>
if mylist:
    # Do something with my list
else:
    # The list is empty
</code>

===== 遍历 list 的同时获取索引 =====


传统的方式：
<code python>
i = 0
for element in mylist:
    # Do something with i and element
    i += 1
</code>

这样更简洁些：
<code python>
for i, element in enumerate(mylist):
    # Do something with i and element
    pass
</code>

===== list 排序 =====


在包含某元素的列表中依据某个属性排序是一个很常见的操作。例如这里我们先创建一个包含 person 的 list：
<code python>
class Person(object):
    def __init__(self, age):
        self.age = age

persons = [Person(age) for age in (14, 78, 42)]
</code>

传统的方式是：

<code python>
def get_sort_key(element):
    return element.age

for element in sorted(persons, key=get_sort_key):
    print "Age:", element.age
</code>

更加简洁、可读性更好的方法是使用 Python 标准库中的 operator 模块：

<code python>
from operator import attrgetter

for element in sorted(persons, key=attrgetter('age')):
    print "Age:", element.age
</code>

attrgetter 方法优先返回读取的属性值作为参数传递给 sorted 方法。operator 模块还包括 itemgetter 和 methodcaller 方法，作用如其字面含义。

===== 在 Dictionary 中元素分组 =====


和上面类似，先创建 Persons：
<code python>
class Person(object):
    def __init__(self, age):
        self.age = age

persons = [Person(age) for age in (78, 14, 78, 42, 14)]
</code>

如果现在我们要按照年龄分组的话，一种方法是使用 in 操作符：
<code python>
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
</code>

相比较之下，使用 collections 模块中 defaultdict 方法的途径可读性更好：
<code python>
from collections import defaultdict

persons_by_age = defaultdict(list)

for person in persons:
    persons_by_age[person.age].append(person)
</code>

defaultdict 将会利用接受的参数为每个不存在的 key 创建对应的值，这里我们传递的是 list，所以它将为每个 key 创建一个 list 类型的值。

===== 参考 =====

  * 内容来源：[[http://alexmarandon.com/articles/python_collections_tips/|AlexMarandon.com]]
  * http://linuxtoy.org/archives/python-collection-tips.html
Python 俱乐部

用户工具

站点工具

页面工具
