python 一次遍历多个迭代对象

这里的 a,b 可以是任何迭代对象

a = [1,3,5,7,9 ]
b = (0,2,4,6,8)
for i in a:
  print(i)  #1 3 5 7 9
for i in b:
  print(i)  # 0 2 4 6 8

使用itertools模块中的chain函数优化以上代码

from itertools import chain
for i in chain(a,b):
  print(i)  #1 3 5 7 9 0 2 4 6 8

你可能觉得先把a和b合并起来再做处理：

a.extend(b)
for i in a:
   print(i) #1 3 5 7 9 0 2 4 6 8

但是会改变原来的a，所以直接用chain

看完记得点赞哦，笔芯