1.13 通过关键字排序字典列表

问题描述

你有一个字典列表，你想根据某个或某几个字典字段来排序这个列表。

解决方案

通过使用operator模块的itemgetter函数，可以实现这样的排序。例如：

from operator import itemgetter

rows = [
    {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003},
    {'fname': 'David', 'lname': 'Beazley', 'uid': 1002},
    {'fname': 'John', 'lname': 'Cleese', 'uid': 1001},
    {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}
]

rows_by_fname = sorted(rows, key=itemgetter('fname'))
rows_by_uid = sorted(rows, key=itemgetter('uid'))
print(rows_by_fname)
"""输出结果
[{'fname': 'Big', 'lname': 'Jones', 'uid': 1004}, 
 {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003}, 
 {'fname': 'David', 'lname': 'Beazley', 'uid': 1002}, 
 {'fname': 'John', 'lname': 'Cleese', 'uid': 1001}]
"""
print(rows_by_uid)
"""输出结果
[{'fname': 'John', 'lname': 'Cleese', 'uid': 1001},
 {'fname': 'David', 'lname': 'Beazley', 'uid': 1002}, 
 {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003}, 
 {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}]
"""

# itemgetter()函数也支持多个keys
rows_by_lfname = sorted(rows, key=itemgetter('lname', 'fname'))
print(rows_by_lfname)
"""输出结果
[{'fname': 'David', 'lname': 'Beazley', 'uid': 1002}, 
 {'fname': 'John', 'lname': 'Cleese', 'uid': 1001}, 
 {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}, 
 {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003}]
"""

讨论

在上面的例子中，字典列表rows被传递给接受一个关键字参数的内置函数sorted()。这个参数是callable类型，并且从rows中接受一个单一元素，然后返回被用来排序的值。itemgetter()函数就是负责创建这个callable对象的。

operator.itemgetter()函数有一个索引参数，可以被rows用来查找值。它可以是一个字典键名称，一个整型值或者任何能够传入一个对象的__getitem__()方法的值。如果你传入多个索引参数给itemgetter()，它生成的callable对象会返回一个包含所有元素值的元组，并且sorted()函数会根据这个元组中元素顺序去排序。

itemgetter()也可以用lambda代替：

rows_by_fname = sorted(rows, key=lambda r: r['fname'])
rows_by_lfname = sorted(rows, key=lambda r: (r['lname'],r['fname']))

但是使用itemgetter()会运行的更快些。如果你对性能有要求，就是用itemgetter()。

当然，这也同样适用于min()和max()等函数。比如：

min(rows, key=itemgetter('uid'))  # {'fname': 'John', 'lname': 'Cleese', 'uid': 1001}
max(rows, key=itemgetter('uid'))  # {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}