kalor

导航

 

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'}]  
38  

 

  

posted on 2014-04-06 22:00  kalor  阅读(6629)  评论(0编辑  收藏  举报