1. 程式人生 > >Python中的sorted函式以及operator.itemgetter函式 【轉載】

Python中的sorted函式以及operator.itemgetter函式 【轉載】

operator.itemgetter函式
operator模組提供的itemgetter函式用於獲取物件的哪些維的資料,引數為一些序號(即需要獲取的資料在物件中的序號),下面看例子。

a = [1,2,3] 
>>> b=operator.itemgetter(1)      //定義函式b,獲取物件的第1個域的值
>>> b(a) 

>>> b=operator.itemgetter(1,0)  //定義函式b,獲取物件的第1個域和第0個的值
>>> b(a) 
(2, 1)

要注意,operator.itemgetter函式獲取的不是值,而是定義了一個函式,通過該函式作用到物件上才能獲取值。

sorted函式
Python內建的排序函式sorted可以對list或者iterator進行排序,官網文件見:http://docs.python.org/2/library/functions.html?highlight=sorted#sorted,該函式原型為:

sorted(iterable[, cmp[, key[, reverse]]])

引數解釋:

(1)iterable指定要排序的list或者iterable,不用多說;

(2)cmp為函式,指定排序時進行比較的函式,可以指定一個函式或者lambda函式,如:

      students為類物件的list,沒個成員有三個域,用sorted進行比較時可以自己定cmp函式,例如這裡要通過比較第三個資料成員來排序,程式碼可以這樣寫:
      students = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
      sorted(students, key=lambda student : student[2])
(3)key為函式,指定取待排序元素的哪一項進行排序,函式用上面的例子來說明,程式碼如下:
      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))

即先根據第二個域排序,再根據第三個域排序。
(4)reverse引數就不用多說了,是一個bool變數,表示升序還是降序排列,預設為false(升序排列),定義為True時將按降序排列。