operator.itemgetter()函数的用法

单独使用operator.itemgetter()函数

• 使用这个函数，需要先调用operator包。
• 此函数用于获取对象的哪些维的数据，参数为一些序号。

例1：

a = np.random.randint(0, 100, (5, 1, 5))
b = operator.itemgetter(1) # 定义函数b，获取对象的第1个域的值
print(a)
print(b(a))

结果为：

[[[ 3 58 52 99 49]]

 [[ 8 79 42 66 16]]

 [[34 16 39 82 72]]

 [[ 7 25 97 12 67]]

 [[78 52 84 33 63]]]
[[ 8 79 42 66 16]]

例二：

a = np.random.randint(0, 100, (5, 1, 5))
b = operator.itemgetter(1, 2) 
print(a)
print(b(a))

[[[65 24 92 13 16]]

 [[33 63 17 13 66]]

 [[99 85 30 89 79]]

 [[20 98 33 92 49]]

 [[18  2 86 40 27]]]
(array([[33, 63, 17, 13, 66]]), array([[99, 85, 30, 89, 79]]))

与sorted()函数一起使用

例1：

a = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
print(a)
c = sorted(a, key=operator.itemgetter(2))
print(c)

[('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]

此例演示的是按照第三位数字进行排序。

例2：

a = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
print(a)
c = sorted(a, key=operator.itemgetter(1, 2))
print(c)

[('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
[('john', 'A', 15), ('dave', 'B', 10), ('jane', 'B', 12)]

此例演示的是先按照第二位字母，再按照第三位数字进行排序。