单独使用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)]
此例演示的是先按照第二位字母,再按照第三位数字进行排序。