問題:根據某個或某幾個字典字段來排序Python列表
answer: 通過使用operator 模塊的itemgetter 函數,可以非常容易的排序這樣的數據結構
eg:
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}
]
1.itemgetter()
from operator import itemgetter
1).print(rows_by_fname)
>>>rows_by_fname = sorted(rows, key=itemgetter('fname'))
>>>print(rows_by_fname)
[{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
{'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'},
{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'}]
2).print(rows_by_uid)
>>>rows_by_uid = sorted(rows, key=itemgetter('uid'))
>>>print(rows_by_uid)
[{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'},
{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
{'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'},
{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'}]
3).itemgetter() 函數也支持多個keys
itemgetter() 函數也支持多個keys,比如下面的代碼
>>>rows_by_lfname = sorted(rows, key=itemgetter('lname','fname'))
>>>print(rows_by_lfname)
[{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'},
{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
{'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'}]
在上面例子中,rows被傳遞給接受一個關鍵字參數的sorted()內置函數。
這個參數是callable類型,並且從rows中接受一個單一元素,然后返回被用來排序的值。
itemgetter()就是負責創建這個callable對象的。
如果傳多個參數給itemgetter(),它生成的callabe對象會返回一個包含所有值的元組,sort()函數會根據元組中元素順序去排序。
2.lambda表達式
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() 方式。
3.min(),max()
最后,不要忘了這節中展示的技術也同樣適用於min() 和max() 等函數。比如:
>>> min(rows, key=itemgetter('uid'))
{'fname': 'John', 'lname': 'Cleese', 'uid': 1001}
>>> max(rows, key=itemgetter('uid'))
{'fname': 'Big', 'lname': 'Jones', 'uid': 1004}