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

問題

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

解法

通過使用 operator 模塊的 itemgetter 函數(shù)，可以非常容易的排序這樣的數(shù)據(jù)結(jié)構(gòu)。假設(shè)你從數(shù)據(jù)庫(kù)中檢索出來(lái)網(wǎng)站會(huì)員信息列表，并且以下列的數(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ù)任意的字典字段來(lái)排序輸入結(jié)果行是很容易實(shí)現(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ù)也支持多個(gè) keys，比如下面的代碼

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

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

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

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

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

這種方案也不錯(cuò)。但是，使用 itemgetter() 方式會(huì)運(yùn)行的稍微快點(diǎn)。因此，如果你對(duì)性能要求比較高的話就使用 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} >>>

總結(jié)

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（1.13）：通过某个关键字排序一个字典列表的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。