python itemgetter函數(shù)用法是什么

小億
86
2024-08-23 02:14:30

itemgetter函數(shù)是python標(biāo)準(zhǔn)庫(kù)中的一個(gè)函數(shù)，位于operator模塊中。它用于獲取對(duì)象的指定索引或鍵對(duì)應(yīng)的值。

itemgetter函數(shù)的用法如下：

from operator import itemgetter

# 創(chuàng)建一個(gè)字典列表
data = [{'name': 'Alice', 'age': 30},
        {'name': 'Bob', 'age': 25},
        {'name': 'Charlie', 'age': 35}]

# 使用itemgetter獲取字典列表中每個(gè)字典的'name'鍵對(duì)應(yīng)的值
get_name = itemgetter('name')
names = list(map(get_name, data))
print(names)  # ['Alice', 'Bob', 'Charlie']

# 使用itemgetter獲取字典列表中每個(gè)字典的指定索引對(duì)應(yīng)的值
get_age = itemgetter(1)
ages = list(map(get_age, data))
print(ages)  # [30, 25, 35]

在上面的例子中，我們通過(guò)itemgetter函數(shù)分別獲取了字典列表中每個(gè)字典的’name’鍵和指定索引對(duì)應(yīng)的值，并將這些值保存在一個(gè)列表中。這樣我們就可以很方便地對(duì)字典列表中的數(shù)據(jù)進(jìn)行提取和操作。

0 踩