高质量Python迭代技巧：enumerate()和zip()

作者：牵引小哥

公众号：牵引小哥讲Python

本期小哥将简单讲解提高Python代码质量的两个迭代技巧。

1. enumerate()代替range()

假设有一个食物的字符串迭代列表foods，需要给它编号，并输出对应的食物food。最简单常规的「笨方法」是这样：

# 笨方法
foods = ['apple', 'banana', 'lemon', 'milk', 'bread']
for i in range(len(foods)):
    print(f'食物编号 {i} 对应的食物是：{foods[i]}')

运行代码得到：

食物编号 0 对应的食物是：apple
食物编号 1 对应的食物是：banana
食物编号 2 对应的食物是：lemon
食物编号 3 对应的食物是：milk
食物编号 4 对应的食物是：bread

这种方法分别用到了range()和len()函数。最后输出还需要根据索引去访问「foods」中的字符串。无论是看起来还是做起来都有点「复杂！」

那么，在这里小哥推荐使用enumerate()函数。先来看看效果：

# 高级方法，使用 enumerate()代替 range()
for i, food in enumerate(foods):
    print(f'食物编号 {i} 对应的食物是：{food}')

运行代码得到：

食物编号 0 对应的食物是：apple
食物编号 1 对应的食物是：banana
食物编号 2 对应的食物是：lemon
食物编号 3 对应的食物是：milk
食物编号 4 对应的食物是：bread

从结果不难看出，使用enumerate()函数的代码更加简洁。

• enumerate() 函数用于将一个可遍历的数据对象组合为一个索引序列，同时列出数据和数据下标。并且可以指定下标开始参数。

如果需要食物编号从1开始，那么可以这样写：

for i, food in enumerate(foods, 1):
    print(f'食物编号 {i} 对应的食物是：{food}')

运行代码得到：

食物编号 1 对应的食物是：apple
食物编号 2 对应的食物是：banana
食物编号 3 对应的食物是：lemon
食物编号 4 对应的食物是：milk
食物编号 5 对应的食物是：bread

2. zip()

如果有一群小朋友，他们分别喜欢不同的水果，需要输出每一个小朋友和他对应的水果。这里涉及两个迭代内容，如果只在一层循环中实现，可使用zip()函数。

• zip() 函数用于将可迭代的对象作为参数，将对象中对应的元素打包成一个个元组，然后返回由这些元组组成的列表。

foods = ['apple', 'banana', 'lemon', 'milk', 'bread']
children = ['小赵', '小钱', '小孙', '小李', '小杨']

for child, food in zip(children, foods):
    print(f'{child} 最喜欢的食物是 {food}')

运行代码得到：

小赵 最喜欢的食物是 apple
小钱 最喜欢的食物是 banana
小孙 最喜欢的食物是 lemon
小李 最喜欢的食物是 milk
小杨 最喜欢的食物是 bread

可使用list()输出组合的列表：

list(zip(children, foods))

运行代码得到：

[('小赵', 'apple'),
 ('小钱', 'banana'),
 ('小孙', 'lemon'),
 ('小李', 'milk'),
 ('小杨', 'bread')]

长按扫码关注

一起学Python

------------------


原创不易，莫要白嫖，如果觉得有点用的话，请毫不留情地素质三连吧，分享、点赞、在看、收藏，我不挑，因为这将是我写作更多优质文章的最强动力。
往期精选（????猛戳可查看）
1、厌倦了if.else的你，能否考虑换一种选择结构？

2、5分钟教你get数据指标体系

3、整理了我开始分享学习笔记到现在超过250篇优质文章，涵盖数据分析、爬虫、机器学习等方面，别再说不知道该从哪开始，实战哪里找了