itemgetter()函数是Python中的一个内置函数,它的作用是从一个可迭代对象(如列表、元组、字典等)中取出一个或多个元素。它可以通过索引或属性名来获取元素,返回值是一个元组。
使用方法:
- 使用itemgetter()函数时,需要传入一个参数,可以是一个索引,也可以是一个属性名。
- 如果参数是一个索引,则会返回该索引对应的元素;如果参数是一个属性名,则会返回该属性名对应的元素。
- 如果传入的参数是一个列表,则会返回一个元组,元组中的每个元素对应列表中的每个索引或属性名。
# 使用itemgetter()函数从字典中获取元素 # 定义一个字典 mydict = {'name': 'John', 'age': 27, 'gender': 'male'} # 使用itemgetter()函数从字典中获取元素 from operator import itemgetter # 传入单个参数 name = itemgetter('name')(mydict) print(name) # 输出:John # 传入多个参数 name_age = itemgetter('name', 'age')(mydict) print(name_age) # 输出:('John', 27)