operator.itemgetter函数

Posted 2020-10-10 子彦

operator.itemgetter函数

operator模块提供的itemgetter函数用于获取对象的哪些维的数据，参数为一些序号（即需要获取的数据在对象中的序号）,要注意，operator.itemgetter函数获取的不是值，而是定义了一个函数，通过该函数作用到对象上才能获取值。

eg:

 students = [(‘john‘, ‘A‘, 15), (‘jane‘, ‘B‘, 12), (‘dave‘, ‘B‘, 10)]

 sorted(students, key=lambda student : student[2])

 

key指定的lambda函数功能是去元素student的第三个域（即：student[2]），因此sorted排序时，会以students所有元素的第三个域来进行排序。有了上面的operator.itemgetter函数，也可以用该函数来实现，例如要通过student的第三个域排序，可以这么写：

         sorted(students, key=operator.itemgetter(2)) 

sorted函数也可以进行多级排序，例如要根据第二个域和第三个域进行排序，可以这么写：

         sorted(students, key=operator.itemgetter(1,2)) 

 1 #-*- encoding=utf-8 -*-  
 2   
 3 import operator  
 4 #按字典值排序（默认为升序）  
 5 x = {1:2, 3:4, 4:3, 2:1, 0:0}  
 6 sorted_x = sorted(x.iteritems(), key=operator.itemgetter(1))  
 7 print sorted_x  
 8 #[(0, 0), (2, 1), (1, 2), (4, 3), (3, 4)]  
 9 #如果要降序排序,可以指定reverse=True  
10 sorted_x = sorted(x.iteritems(), key=operator.itemgetter(1), reverse=True)  
11 print sorted_x  
12 #[(3, 4), (4, 3), (1, 2), (2, 1), (0, 0)]  
13 #或者直接使用list的reverse方法将sorted_x顺序反转  
14 #sorted_x.reverse()  
15   
16 #取代方法是,用lambda表达式  
17 sorted_x = sorted(x.iteritems(), key=lambda x : x[1])  
18 print sorted_x  
19 #[(0, 0), (2, 1), (1, 2), (4, 3), (3, 4)]  
20 sorted_x = sorted(x.iteritems(), key=lambda x : x[1], reverse=True)  
21 print sorted_x  
22 #[(3, 4), (4, 3), (1, 2), (2, 1), (0, 0)]  
23   
24 #包含字典dict的列表list的排序方法与dict的排序类似,如下：  
25 x = [{‘name‘:‘Homer‘, ‘age‘:39}, {‘name‘:‘Bart‘, ‘age‘:10}]  
26 sorted_x = sorted(x, key=operator.itemgetter(‘name‘))  
27 print sorted_x  
28 #[{‘age‘: 10, ‘name‘: ‘Bart‘}, {‘age‘: 39, ‘name‘: ‘Homer‘}]  
29 sorted_x = sorted(x, key=operator.itemgetter(‘name‘), reverse=True)  
30 print sorted_x  
31 #[{‘age‘: 39, ‘name‘: ‘Homer‘}, {‘age‘: 10, ‘name‘: ‘Bart‘}]  
32 sorted_x = sorted(x, key=lambda x : x[‘name‘])  
33 print sorted_x  
34 #[{‘age‘: 10, ‘name‘: ‘Bart‘}, {‘age‘: 39, ‘name‘: ‘Homer‘}]  
35 sorted_x = sorted(x, key=lambda x : x[‘name‘], reverse=True)  
36 print sorted_x  
37 #[{‘age‘: 39, ‘name‘: ‘Homer‘}, {‘age‘: 10, ‘name‘: ‘Bart‘}]