Python减少代码量的两个内置函数

咪哥杂谈

本篇阅读时间约为 5 分钟。

1

前言

Python中内置了几个非常好用的函数。

当你掌握了这几个函数的用法后，有些场景下，不用自己去实现多余的冗余代码编写，只需要调用这些函数，便能很简短的帮你实现功能。

今天来分享下两个函数的场景以及用法。

PS:下例子截图演示均在 jupyter notebook 中完成。

2

enumerate

enumerate，单词是枚举的意思。

枚举是什么意思呢？来看下 Java 中的枚举，直接选自百度提问。这个例子举得比较生动。

在你理解了枚举的意思后，来看下 Python 中枚举函数的使用场景。

场景：

在 Python 中，我们想去遍历一个 list 列表，有哪种方式可以将列表中的元素全部打印出来呢？

假设现在有个 list ，其中包含了 1 - 5 五个数字，用程序将每个元素打印出来。

number_list = [1,2,3,4,5]

自行思考下，再继续往后看。

方法一，直接 for 循环：

for num in number_list:    print(num)

方法二， for + range() + len()：

for i in range(len(number_list)):    print(number_list[i])

不管哪种方式，你最终都能得到如下结果：

如果，现在让你将列表中的每个元素和它对应的索引位置打印出来，如何做到？依然是自行思考下，再往后看。

是不是直接用第一种方法 for...in list 的方式，显然不能完成。

而第二种方法可以完成，只需要如下：

但是这种方式是否觉得非常麻烦？你写的代码既套用 range ，又需要套用 len，过于冗长。

此时，Python 内置函数 enumerate 完美的解决了这个场景下问题。

for i,num in enumerate(number_list):    print(f'当前索引位置:{i}')    print(f'当前元素:{num}')

对比下两种方式，相信你会爱上第二种写法：

用 enumerate 函数，遍历出来的第一个变量是索引位置，第二个是结果，这二者顺序不可颠倒，第一个变量一定是 int 类型的数字。至于变量的名字，想叫什么叫什么，没有影响。

例子中 i，num 。你也可以替换成别的。

记住场景，就容易想到 enumerate 函数了，当你遍历序列时，既想要序列对应索引位置，又想要序列中的内容，选它，enumerate() ！

还记得序列的概念吗？就是能遍历的类型。

3

zip

看到 zip 这个单词时，但凡用过压缩包的同学都不陌生吧。

我们经常会看到，许多文件都是 xxx.zip 的文件格式，即压缩包。

有了上述理解，来看下 zip 的场景。

场景：

有个男士列表，里面分别是 黄晓明，刘恺威，贾乃亮。

有个女士列表，里面分别是 杨颖，杨幂，李小璐。

能看出来，男士列表和女士列表，都是按照顺序配对写的。

如果将两个 list 中的对应元素，将他们取出来组合成夫妻呢？

man_list = ['黄晓明','刘恺威','贾乃亮']woman_list = ['杨颖','杨幂','李小璐']

想要的结果：

[('黄晓明','杨颖'),('刘恺威','杨幂'),('贾乃亮','李小璐')]

自行思考，正常我们需要怎么做，才能实现？

正常逻辑：

for i in range(len(man_list)):    couple_list.append((man_list[i],woman_list[i]))print(couple_list)

遍历第一个元素列表，获取到索引的同时，将男女列表此时的索引元素组成元组，放入到 couple_list 中。

看下我们的 zip 函数：

zip可以将多个列表、元组或其它序列成对组合成一个元组列表。

所以，你只需要完美的将两个元素传入，最终转为 list 即可：

couple_zip_list = list(zip(man_list,woman_list))print(couple_zip_list)

什么叫做简洁？一行代码搞定，有木有觉得就是这么 easy 啊！

别急，正常场景，一般不会这么巧的给你两个同等元素列表，假设还有另外一个列表，我们将其定义为是否有孩子列表吧。

关注娱乐的朋友都知道，其实这三对夫妻都已经有孩子了，分别叫小海绵，小糯米，贾云馨。

man_list = ['黄晓明','刘恺威','贾乃亮']woman_list = ['杨颖','杨幂','李小璐']has_child = [True,True]

现在有三个列表，依然是顺序打包组合。最后来猜猜结果会如何呢？

看到，结果没有最后一对夫妻的结果了。。。

zip虽然可以处理任意多的序列，但最终元素的个数取决于最短的序列。

逆向思维一下，如果 zip 能正向打包，是否可以逆向解包呢？

答案，当然是可以！zip解包 确实很神奇。

man_tuple,woman_tuple = zip(*couple_list)

注意，解包出来的类型是 tuple 而不是 list。

4

组合打法

上述介绍的两个函数，可以结合使用，形成终极大招：

场景：

假设三人同时在一场婚礼上举办。

man_list = ['黄晓明', '刘恺威', '贾乃亮']woman_list = ['杨颖', '杨幂', '李小璐']

此时牧师要一次宣讲，第几对新人，男方是谁，女方是谁。。打印如下：

当前第1对夫妻: 男方：黄晓明, 女方：杨颖

所以，结合场景，组合今天介绍的 enumerate 和 zip 这样使用，就能让牧师完美的讲出来。

for i, (man, woman) in enumerate(zip(man_list, woman_list)):    print(f'当前第{i+1}对夫妻: 男方：{man}, 女方：{woman}')

5

总结

好了，简单的回顾总结一下今天的内容。

第一个 enumerate 函数的场景，什么时候会用到？当你既想要序列中的索引坐标，又想要内容的时候！注意顺序。

第二个 zip 函数的场景，什么时候会用到？当你需要对多个序列同时进行遍历的时候，用 zip 会省不少事！

特定场景下，结合两种方法使用，可以达到神奇的效果。

总之，这两种内置方法从使用角度来讲，大大的降低了代码编写的复杂度，非常友好。

关于今天分享的内容就到这里啦，你学会了嘛？有什么问题，可以评论区留言探讨呐！

---

▼

往期精彩回顾

▼

大家好，我，一岁啦！
50行代码爬取穷游网

爬虫神器之 PyQuery 实用教程（一）

你点的每个在看，我都认真当成了喜欢