Python operator.itemgetter

实例

我们先声明一个列表，里面元素为字典

data = [
    {'age': 31, 'city': 'taipei', 'name': 'amy'},
    {'age': 71, 'city': 'tokyo', 'name': 'john'},
    {'age': 16, 'city': 'london', 'name': 'zoe'},
    {'age': 16, 'city': 'rio', 'name': 'cathy'},
    {'age': 48, 'city': 'frankfurt', 'name': 'david'}]

排序

我们用sorted对年龄进行排序，参数key传入lambda，指定要根据哪个元素进行排序

print(sorted(data, key=lambda x: x['age']))

输出：

[{'age': 16, 'city': 'london', 'name': 'zoe'},
 {'age': 16, 'city': 'rio', 'name': 'cathy'},
 {'age': 31, 'city': 'taipei', 'name': 'amy'},
 {'age': 48, 'city': 'frankfurt', 'name': 'david'},
 {'age': 71, 'city': 'tokyo', 'name': 'john'}]

其实，key用operator.itemgetter指定想要排序的元素，并且速度更快

from operator import itemgetter
data = [
    {'age': 31, 'city': 'taipei', 'name': 'amy'},
    {'age': 71, 'city': 'tokyo', 'name': 'john'},
    {'age': 16, 'city': 'london', 'name': 'zoe'},
    {'age': 16, 'city': 'rio', 'name': 'cathy'},
    {'age': 48, 'city': 'frankfurt', 'name': 'david'}]
print(sorted(data, key=itemgetter('age')))

结果和上面一样

[{'age': 16, 'city': 'london', 'name': 'zoe'},
 {'age': 16, 'city': 'rio', 'name': 'cathy'},
 {'age': 31, 'city': 'taipei', 'name': 'amy'},
 {'age': 48, 'city': 'frankfurt', 'name': 'david'},
 {'age': 71, 'city': 'tokyo', 'name': 'john'}]