【Python基础教程】在Python for语句中迭代多个可迭代对象的方法

╰+哭是因爲堅強的太久メ 2022-09-12 14:59 169阅读 0赞

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循环引用

''' 学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：531509025 寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！ '''
    #弊端：只能支持引索操作：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，它能使多个迭代对象连接

场景一：

''' 学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：531509025 寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！ '''
    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)