1. 程式人生 > >python中的operator.itemgetter函式

python中的operator.itemgetter函式

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

a = [1,2,3]

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

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

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

sorted函式用來排序,sorted(iterable[, cmp[, key[, reverse]]])

其中key的引數為一個函式或者lambda函式。所以itemgetter可以用來當key的引數

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

根據第二個域和第三個域進行排序

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