operator itemgetter 使用

operator模块

能替代从序列中取出元素或读取对象属性的 lambda 表达式： itemgetter 和 attrgetter 其实会自行构建函数。 作用 根据元组的某个字段给元组列表排序。itemgetter(1) 的作用与 lambda fields: fields[1] 一样： 创建一个接受集合的函数， 返回索引位 1 上的元素

序列案例

metro_data = [
    ('Tokyo', 'JP', 36.933, (35.689722, 139.691667)),
    ('Delhi NCR', 'IN', 21.935, (28.613889, 77.208889)),
    ('Mexico City', 'MX', 20.142, (19.433333, -99.133333)),
    ('New York-Newark', 'US', 20.104, (40.808611, -74.020386)),
    ('Sao Paulo', 'BR', 19.649, (-23.547778, -46.635833)),
]

代码

from operator import itemgetter

# 单个参数
for item in sorted(metro_data, key=itemgetter(1))
    print(item)
# 等同下方
for item in sorted(metro_data, key=lambda x: x[1]):
    print(item)
# 多个参数

cc_name = itemgetter(1, 0)
for item in metro_data:
    print(cc_name(item))

itemgetter 使用 [] 运算符， 因此它不仅支持序列， 还支持映射和任何实现 getitem 方法的类

映射案例

代码

from random import randint

map_data = [{'id': randint(1, 5), 'age': randint(16, 18)} for _ in range(10)]

# lambda
sorted(map_data, key=lambda x: (x['id'], x['age']))
# itemgetter
sorted(map_data, key=itemgetter('id', 'age'))
# 都可以完成 序列map_data首选根据id排序。id相同根据age 排序。reverse 默认为false 升序

映射案例是一次 北京 python后端开发工程师面试题。这里提供了两个方法