码迷,mamicode.com
首页 > 其他好文 > 详细

问题13:如何在for语句中迭代多个可迭代的对象

时间:2018-04-09 13:23:06      阅读:198      评论:0      收藏:0      [点我收藏+]

标签:dom   color   并行迭代   标准库   .com   pos   标准   tle   for循环   

from random import randint
a1 = [randint(10, 50) for _ in range(5)]
a2 = [randint(10, 50) for _ in range(5)]
a3 = [randint(10, 50) for _ in range(5)]
a4 = []

例一:并行操作:在一个for循环中实现多个列表的并行迭代;

方案:使用内置函数zip,将多个迭代对象合并,每次迭代返回一个元组

案例:对3个列表同时迭代,计算各列表对应元素的和;

#方法一:直接用for循环引用
#弊端:只能支持引索操作:a1[],若操作对象是生成器,则不能实现;
for i in range(5):
    t = a1[i] + a2[i] + a3[i]
    a4.append(t)

print(a4)
#输出:[84, 67, 85, 88, 82]

#方法二:用内置函数zip()
for x, y, z in zip(a1, a2, a3):
    a4.append(x + y + z)

print(a4)
#输出:[44, 72, 73, 94, 130]

 

例二:川行操作:在一个for循环中实现多个列表的川行迭代;

方案:使用标准库itertools.chain,它能使多个迭代对象连接

itertools.chain的使用,也可参考:Python:itertools库的使用

场景一:
from
itertools import chain b1 = [1, 2, 3, 4] b2 = [a, b, c] b3 = list(chain(b1, b2)) print(b3) #输出:[1, 2, 3, 4, ‘a‘, ‘b‘, ‘c‘]
场景二: for x in chain(b1, b2): print(x) #输出:1 2 3 4 a b c

 

案例:对4个列表进行迭代操作,筛选出目标数据(大于40的个数):

from itertools import chain
from random import randint
a1 = [randint(10, 50) for _ in range(40)]
a2 = [randint(10, 50) for _ in range(41)]
a3 = [randint(10, 50) for _ in range(42)]
a4 = [randint(10, 50) for _ in range(43)]
count = 0

for x in chain(a1, a2, a3, a4):
    if x >=40:
        count += 1

print(count)

 

问题13:如何在for语句中迭代多个可迭代的对象

标签:dom   color   并行迭代   标准库   .com   pos   标准   tle   for循环   

原文地址:https://www.cnblogs.com/volcao/p/8758910.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!