温馨提示×

python itemgetter函数用法是什么

小亿
86
2024-08-23 02:14:30
栏目: 编程语言

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’键和指定索引对应的值,并将这些值保存在一个列表中。这样我们就可以很方便地对字典列表中的数据进行提取和操作。

0