- 通过某个关键字排序一个字典列表
- 问题:
- 你有一个字典列表,你想根据某个或某几个字典字段来排序这个列表。
- 解决方案:
- 通过使用 operator 模块中的 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}
]
- 根据任意字典字段来排序输入结果行是很容易实现的,代码演示:
from operator import itemgetter
rows_by_fname = sorted(rows, key= itemgetter("fname"))
rows_by_uid = sorted(rows, key= itemgetter("uid"))
print(rows_by_fname)
print(rows_by_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_fname = sorted(rows, key= lambda r:r["fname"])
rows_lfname = sorted(rows, key= lambda r:(r["lname"],r["fname"]))
- 这种方案也不错。但是,使用 itemgetter() 方式会运行的稍微快点。因此,如果对性能要求比较搞的话,就 使用 itemgetter()方式。
- 最后,这节中演示的方法也同样使用于min()和max()等函数。比如:
min(rows, key= itemgetter("uid"))
max(rows, key= itemgetter("uid"))