问题描述
你有一个字典列表,你想根据某个或某几个字典字段来排序这个列表。
解决方案
通过使用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}
来源:https://www.cnblogs.com/L999C/p/15739325.html
图文来源于网络,如有侵权请联系删除。