Python中sorted方法与operator.itemgetter方法的共用

Python中的sorted函数和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)

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

sorted函数

Python内置的排序函数sorted可以对iterator进行排序,可参加官网文档,该函数原型为:

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

参数解释:

  1. iterable:指要排序的 list 或者 iterable;
  2. cmp:是指定排序时进行比较的函数,可以指定一个函数或者lambda函数。
  3. key:指定取待排序元素的哪一项进行排序。

举例来说,students 是一个 list,用sorted进行比较时可以自己定cmp函数,例如这里要通过比较每个元素的第三个值来排序,代码可以这样写:

students = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]

sorted(students, key=lambda student : student[2])

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

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

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

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

即先跟句第二个域排序,再根据第三个域排序。

  1. reverse参数是一个bool变量,表示升序还是降序排列,默认为False(升序排列),定义为True时将按降序排列。

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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 文:郑元春一个有着美好梦想的coder! 人生苦短,我用Python。 P.S. 个人认为,查找和排序是算法的核心...
    北静王阅读 2,162评论 3 10
  • 之前在Udemy上跟着Jose Portilla老师过了一遍Python+Django的教程,做了一个小博客。但是...
    greatseniorsde阅读 458评论 0 0
  • 一声轻唤,说走就走。到山里走走,落山的太阳照着你嫣红的脸庞;到水边走走,奔腾的溪水拉住你轻盈的步履。慢慢走...
    冰夫阅读 221评论 0 0
  • 阅读本篇大概需要20分钟 视频编辑分为两部分:Camrea Engine(视频资源采集)和Video Edit E...
    董二千阅读 2,127评论 0 2
  • 忙碌了一天 太阳已落在了地平线 可灯火依旧阑珊 而照片还在 人说 相遇是缘起 相知是缘始 却总无法答到相守的缘满 ...
    浙江小王子食品阅读 179评论 0 1