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’鍵和指定索引對應的值,并將這些值保存在一個列表中。這樣我們就可以很方便地對字典列表中的數據進行提取和操作。