itemgetter函数是python标准库中的一个函数,位于operator模块中。它用于获取对象的指定索引或键对应的值。
itemgetter函数的用法如下:
from operator import itemgetter
# 创建一个字典列表
data = [{'name': 'Alice', 'age': 30},
{'name': 'Bob', 'age': 25},
{'name': 'Charlie', 'age': 35}]
# 使用itemgetter获取字典列表中每个字典的'name'键对应的值
get_name = itemgetter('name')
names = list(map(get_name, data))
print(names) # ['Alice', 'Bob', 'Charlie']
# 使用itemgetter获取字典列表中每个字典的指定索引对应的值
get_age = itemgetter(1)
ages = list(map(get_age, data))
print(ages) # [30, 25, 35]
在上面的例子中,我们通过itemgetter函数分别获取了字典列表中每个字典的’name’键和指定索引对应的值,并将这些值保存在一个列表中。这样我们就可以很方便地对字典列表中的数据进行提取和操作。