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