写点什么

像这样操作 Python 列表,能让你的代码更优雅 | pythonic 小技巧

用户头像
AlwaysBeta
关注
发布于: 2021 年 03 月 15 日

写 Python 代码,列表的出镜率是相当高的,伴随列表一起出现的往往就是一大堆 for 循环,这样的代码多了看起来非常不简洁。作为一名 Python 程序员,怎么能忍受呢?


那有没有什么好办法呢?除了列表表达式之外,其实还有一些小技巧来操作列表,可以使代码更简洁,更优雅。下面介绍几个常见的使用场景,分享给大家。


一、列表合并


第一种方式:循环。


>>> a = [1, 2, 3]>>> b = [4, 5, 6]>>> for i in b:...     a.append(i)...>>> a[1, 2, 3, 4, 5, 6]
复制代码

这种方式最不友好了,也不建议使用。


第二种方式:使用 +


>>> a + b[1, 2, 3, 4, 5, 6]
复制代码

第三种方式:使用 extend 关键字。


>>> a.extend(b)>>> a[1, 2, 3, 4, 5, 6]
复制代码

后两种方式明显更加优雅,推荐使用。需要说明的一点是,如果列表很大的话,+ 会比较慢,使用 extend 更好。


二、列表元素去重


使用 set() 对列表元素进行去重。


>>> a = [1, 2, 3, 4, 2, 3]>>> list(set(a))[1, 2, 3, 4]
复制代码

三、列表排序


使用 sort() 或内建函数 sorted() 对列表进行排序。它们之间的区别有两点:


  1. sort() 方法是对原列表进行操作,而 sorted() 方法会返回一个新列表,不是在原来的基础上进行操作。

  2. sort() 是应用在列表上的方法,而 sorted() 可以对所有可迭代的对象进行排序操作。


# sort()>>> a = [1, 2, 3, 4, 2, 3]>>> a.sort()>>> a[1, 2, 2, 3, 3, 4]>>>>>> a = [1, 2, 3, 4, 2, 3]>>> a.sort(reverse=True)>>> a[4, 3, 3, 2, 2, 1]
# sorted()>>> a = [1, 2, 3, 4, 2, 3]>>> sorted(a)[1, 2, 2, 3, 3, 4]>>> a = [1, 2, 3, 4, 2, 3]>>> sorted(a, reverse=True)[4, 3, 3, 2, 2, 1]
复制代码

四、遍历列表的索引和元素对


使用 enumerate() 函数可以同时输出索引和元素值。


>>> a = ['python', 'go', 'java']>>> for i, v in enumerate(a):...     print(i, v)
# output0 python1 go2 java
复制代码

五、查找列表中出现最频繁的元素


使用 max() 函数可以快速查找出一个列表中出现频率最高的某个元素。


>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]>>> b = max(set(a), key=a.count)>>> b4
复制代码

需要说明的一点是,当列表中有两个元素出现的次数相同时,会返回第一个出现的元素。


>>> a = [1, 2]>>> b = max(set(a), key=a.count)>>> b1
复制代码

六、统计列表中所有元素的出现次数


前面的代码给出了出现最频繁的值。如果想要知道列表中所有元素的出现次数,那么可以使用 collections 模块。collections 是 Python 中的一个宝藏模块,它提供了很多特性。Counter 方法正好可以完美解决这个需求。


>>> from collections import Counter>>>>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]>>> Counter(a)Counter({4: 4, 2: 2, 3: 2, 1: 1, 5: 1})
复制代码

七、将两个列表合并为词典


使用 zip() 函数,可以将两个列表合并成字典。


>>> a = ['one', 'tow', 'three']>>> b = [1, 2, 3]>>> dict(zip(a, b)){'one': 1, 'tow': 2, 'three': 3}
复制代码

以上。


希望对你能有帮助,欢迎关注公众号 AlwaysBeta,更多技术干货等你来。



发布于: 2021 年 03 月 15 日阅读数: 19
用户头像

AlwaysBeta

关注

微信公众号:AlwaysBeta 2017.11.30 加入

后端开发,技术栈:Python,Django,Redis,RabbitMQ,Kafka,Elasticsearch,MySQL

评论

发布
暂无评论
像这样操作 Python 列表,能让你的代码更优雅 | pythonic 小技巧