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}]

注意字典必须包含 itemgetter 中指定的所有键,否则会抛出 KeyError。

小贴士:

  • itemgetter 返回的是函数,适合用在 key 参数中
  • 比 lambda 更高效,尤其是在处理大量数据时
  • 支持负索引,如 itemgetter(-1) 取最后一个元素

基本上就这些,不复杂但容易忽略。