Python中的operator.itemgetter函数

Python中的operator.itemgetter函数

一、总结

一句话总结：

【itemgetter函数用于获取对象的哪些维的数据】：operator模块提供的itemgetter函数用于获取对象的哪些维的数据，参数为一些序号（即需要获取的数据在对象中的序号）

实例：a = [1,2,3] b=operator.itemgetter(2,1) print(b(a)) #结果3,2

【itemgetter函数是定义了一个函数】：要注意，operator.itemgetter函数获取的不是值，而是定义了一个函数，通过该函数作用到对象上才能获取值。

 

1、operator.itemgetter函数在sort函数中的应用？

【key=lambda student : student[2]】：students = [('john', 'C', 15), ('jane', 'B', 17), ('dave', 'B', 10)]  ans=sorted(students, key=lambda student : student[2])

【key=operator.itemgetter(2)】：operator.itemgetter函数，通过student的第三个域排序：ans2=sorted(students, key=operator.itemgetter(2))

【key=operator.itemgetter(1,2)】：operator.itemgetter函数多级排序：ans3=sorted(students, key=operator.itemgetter(1,2))

 

 

二、Python中的operator.itemgetter函数

转自或参考：Python中的sorted函数以及operator.itemgetter函数
https://blog.csdn.net/dongtingzhizi/article/details/12068205

1、operator.itemgetter函数

operator模块提供的itemgetter函数用于获取对象的哪些维的数据，参数为一些序号（即需要获取的数据在对象中的序号），下面看例子。

In [1]:

import operator
a = [1,2,3] 
b=operator.itemgetter(1) #定义函数b，获取对象的第1个域的值
print(b(a))

2

In [2]:

b=operator.itemgetter(1,0) #定义函数b，获取对象的第1个域和第0个的值
print(b(a))

(2, 1)

In [9]:

b=operator.itemgetter(2,1) #定义函数b，获取对象的第2个域和第1个的值
print(b(a))

(3, 2)

要注意，operator.itemgetter函数获取的不是值，而是定义了一个函数，通过该函数作用到对象上才能获取值。

2、operator.itemgetter函数在sort函数中的应用

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时将按降序排列。

sorted函数更多的例子可以参考官网文档：https://wiki.python.org/moin/HowTo/Sorting/。

key为函数，指定取待排序元素的哪一项进行排序，例如要通过比较第三个数据成员来排序

In [5]:

students = [('john', 'C', 15), ('jane', 'B', 17), ('dave', 'B', 10)] 
ans=sorted(students, key=lambda student : student[2])
print(students)
print(ans)

[('john', 'C', 15), ('jane', 'B', 17), ('dave', 'B', 10)]
[('dave', 'B', 10), ('john', 'C', 15), ('jane', 'B', 17)]

有了上面的operator.itemgetter函数，也可以用该函数来实现，例如要通过student的第三个域排序，可以这么写：

In [7]:

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

[('john', 'C', 15), ('jane', 'B', 17), ('dave', 'B', 10)]
[('dave', 'B', 10), ('john', 'C', 15), ('jane', 'B', 17)]

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

In [8]:

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

[('john', 'C', 15), ('jane', 'B', 17), ('dave', 'B', 10)]
[('dave', 'B', 10), ('jane', 'B', 17), ('john', 'C', 15)]

In [ ]: