operator.itemgetter()用法

operator中的itemgetter函数

from operator import itemgetter

• 该函数的作用是获取对象的某个/些维度的数据,默认将对象的维度看为1维
• 该函数的使用方法:
  • 1 . 定义一个itemgetter函数,并指定获取某个对象中哪些维度的值
  • 2 . 用定义好的函数作用于对象
• 示例图下

# 定义1个1维列表和1个二维列表
a = [1,2,3]
b = [[1,2,3],[4,5,6],[7,8,9]]

# 定义itemgetter函数,获取a、b中的第0维的数据
get = itemgetter(0)
print(get(a)) # 返回的是第0为的值
print(get(b)) # 返回的是一个列表,也是其第0为的值

1
[1, 2, 3]

# 定义itemgetter函数,获取a、b中的第0维和第2维的数据
get = itemgetter(0,2)
print(get(a)) # 返回的是元组,其中包含了第0维和第2维的数据
print(get(b)) # 返回的是元组

(1, 3)
([1, 2, 3], [7, 8, 9])