1.13 通過某個關(guān)鍵字排序一個字典列表

問題

你有一個字典列表，你想根據(jù)某個或某幾個字典字段來排序這個列表。

解決方案

通過使用operator模塊的itemgetter函數(shù)，可以非常容易的排序這樣的數(shù)據(jù)結(jié)構(gòu)。假設(shè)你從數(shù)據(jù)庫中檢索出來網(wǎng)站會員信息列表，并且以下列的數(shù)據(jù)結(jié)構(gòu)返回：

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}
]

根據(jù)任意的字典字段來排序輸入結(jié)果行是很容易實現(xiàn)的，代碼示例：

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)

代碼的輸出如下：

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

itemgetter()函數(shù)也支持多個keys，比如下面的代碼

rows_by_lfname = sorted(rows, key=itemgetter('lname','fname'))
print(rows_by_lfname)

會產(chǎn)生如下的輸出：

[{'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 被傳遞給接受一個關(guān)鍵字參數(shù)的 sorted() 內(nèi)置函數(shù)。這個參數(shù)是 callable 類型，并且從 rows 中接受一個單一元素，然后返回被用來排序的值。itemgetter() 函數(shù)就是負(fù)責(zé)創(chuàng)建這個 callable 對象的。

operator.itemgetter() 函數(shù)有一個被rows中的記錄用來查找值的索引參數(shù)?？梢允且粋€字典鍵名稱，一個整形值或者任何能夠傳入一個對象的 __getitem__() 方法的值。如果你傳入多個索引參數(shù)給 itemgetter() ，它生成的 callable 對象會返回一個包含所有元素值的元組，并且sorted()函數(shù)會根據(jù)這個元組中元素順序去排序。但你想要同時在幾個字段上面進(jìn)行排序(比如通過姓和名來排序，也就是例子中的那樣)的時候這種方法是很有用的。

itemgetter() 有時候也可以用lambda表達(dá)式代替，比如：

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

這種方案也不錯。但是，使用itemgetter()方式會運行的稍微快點。因此，如果你對性能要求比較高的話就使用itemgetter()方式。

最后，不要忘了這節(jié)中展示的技術(shù)也同樣適用于min()和max()等函數(shù)。比如：

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