Python itemgetter函数怎么用？

itemgetter用于提取序列或字典中的元素，支持单个或多个索引，常用于排序操作，比lambda更高效，如itemgetter(1)按第二项排序，itemgetter('score')按字典键排序。

itemgetter 是 Python 中 operator 模块提供的一个函数，用来从序列或可迭代对象中提取特定位置的元素。它常用于排序、筛选等操作中，让代码更简洁清晰。

基本用法：提取单个元素

如果你有一个列表的元组，想按某个字段排序，可以用 itemgetter 提取该字段。

from operator import itemgetter
data = [('Alice', 25), ('Bob', 30), ('Charlie', 20)]
按年龄排序（提取第二个元素）
sorted_data = sorted(data, key=itemgetter(1))
print(sorted_data)
输出: [('Charlie', 20), ('Alice', 25), ('Bob', 30)]

这里 itemgetter(1) 返回一个可调用对象，会取出每个元组索引为 1 的元素作为排序依据。

提取多个字段

itemgetter 支持传入多个索引，返回多个值组成的元组。

from operator import itemgetter
data = [('Alice', 25, 'Engineer'), ('Bob', 30, 'Designer'), ('Charlie', 25, 'Teacher')]
先按年龄排序，再按名字排序
sorted_data = sorted(data, key=itemgetter(1, 0))
print(sorted_data)
输出: [('Charlie', 25, 'Teacher'), ('Alice', 25, 'Engineer'), ('Bob', 30, 'Designer')]

当第一个字段相同时，会使用第二个字段进行比较。

用于字典场景

itemgetter 也适用于字典，用键名代替索引。

from operator import itemgetter
students = [
{'name': 'Alice', 'score': 88},
{'name': 'Bob', 'score': 95},
{'name': 'Charlie', 'score': 70}
]
按分数排序
sorted_students = sorted(students, key=itemgetter('score'))
print(sorted_students)
输出: [{'name': 'Charlie', 'score': 70}, {'name': 'Alice', 'score': 88}, {'name': 'Bob', 'score': 95}]