Python operator.itemgetter()



一、問題

對列表、元組、字典或(字典、元組組成的列表),根據屬性值排序。



二、解決方案

1. 列表、元組、字典

operator.itemgetter() 獲取的不是值,而是定義一個函數,通過該函數作用到對象上才能獲取值。

from operator import itemgetter

a = [1, 2, 3]
b = itemgetter(1)       # 定義函數b,獲取對象的第1個值
c = itemgetter(1, 0)    # 定義函數c,獲取對象的第1、0個值
print(b(a))
print(c(a))

a = {'name': 'wangke', 'age':18, 'sex': 'man'}
b = itemgetter('age')       
c = itemgetter('name', 'age')    
print(b(a))
print(c(a))

輸出:

2
(2, 1)

18
('wangke', 18)

2. 字典列表

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_fname = sorted(rows, key=itemgetter('fname'))
rows_uid = sorted(rows, key=itemgetter('uid'))
print(rows_fname)
print(rows_uid)

輸出:

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

[{'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()也可以用lambda代替,但是,itemgetter()會稍微快點。。

rows_fname = sorted(rows, key=lambda r: r['fanme'])
rows_uid = sorted(rows, key=lambda r: (r['fanme'], r['uid']))

3. 元組列表

from operator import itemgetter

a = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'C', 10)]
b = sorted(a, key=itemgetter(1))
c = sorted(a, key=itemgetter(2))
print(b)
print(c)

輸出:

[('john', 'A', 15), ('jane', 'B', 12), ('dave', 'C', 10)]
[('dave', 'C', 10), ('jane', 'B', 12), ('john', 'A', 15)]


三、討論

查找的元素個數相對比較小,用 nsmallest()nlargest()

查找唯一的最大、最小值,用 max()min()

查找元素個數與總的元素個數接近,通常先排序,然后再切片。sorted()




免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM