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))
输出:
2
(2, 1)
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}]
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)]
三、讨论
标签:rows,uid,Python,lname,fname,operator,itemgetter 来源: https://www.cnblogs.com/keye/p/15632092.html