python中的operator.itemgetter函数

时间:2023-03-08 22:28:58

operator.itemgetter函数
operator模块提供的itemgetter函数用于获取对象的哪些维的数据,参数为一些序号。看下面的例子

a = [,,]
>>> b=operator.itemgetter() //定义函数b,获取对象的第1个域的值
>>> b(a) >>> b=operator.itemgetter(,) //定义函数b,获取对象的第1个域和第0个的值
>>> b(a)
(, )

要注意,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))

  • 用 operator 函数进行多级排序
 students = [('john', 'A', ), ('jane', 'B', ), ('dave', 'B', ),]
>>> sorted(students, key=itemgetter(,)) # sort by grade then by age
[('john', 'A', ), ('dave', 'B', ), ('jane', 'B', )]
## 对字典排序
>>> d = {'data1':, 'data2':, 'data3':, 'data4':}
>>> sorted(d.iteritems(), key=itemgetter(), reverse=True)
[('data4', ), ('data1', ), ('data3', ), ('data2', )]