如何在一个for语句中迭代多个可迭代对象

实际案例
  1. 某班学生期末考试成绩,语文、数学和英语分别存储在3个列表中,同时迭代三个列表,计算每个学生的总分。(并行)
  2. 某年级有4个班,某次考试每班英语成绩分别存储在4个列表中,依次迭代每个列表,统计全学年成绩高于90分人数。(串行)

案例一:

我们看完这个案例是不是首先就想起用索引来解决呢?该方法代码如下:

# -*- coding: utf-8 -*-

from random import randint

chinese = [randint(50, 100) for _ in xrange(3)]
math = [randint(50, 100) for _ in xrange(3)]
english = [randint(50, 100) for _ in xrange(3)]

for i in xrange(len(chinese)):
    sum = chinese[i] + math[i] + english[i]
    print i+1, sum

其输出结果为:

1 234
2 245
3 190

大家有没有想过,这种方式有个局限性,就是chinese 、math 和english 这些可迭代对象必须是支持这种索引操作的。如果它们为生成器就不支持索引操作,进而无法解决该问题。

这里(并行情况),我们推荐使用内置函数zip,它能将多个可迭代对象合并,每次返回一个元组。该方法代码如下:

# -*- coding: utf-8 -*-

from random import randint

chinese = [randint(50, 100) for _ in xrange(3)]
math = [randint(50, 100) for _ in xrange(3)]
english = [randint(50, 100) for _ in xrange(3)]

total = []

for c, m, e in zip(chinese, math, english):
    total.append(c+m+e)

for i in xrange(len(chinese)):
    print i+1, total[i]

其输出结果为:

1 234
2 245
3 190

案例二(串行):

解决方案:使用标准库中的itertools.chain,它能将多个可迭代对象连接。该方法代码如下:

# -*- coding: utf-8 -*-

from itertools import chain
from random import randint

e1 = [randint(50, 100) for _ in xrange(3)]
e2 = [randint(50, 100) for _ in xrange(4)]
e3 = [randint(50, 100) for _ in xrange(6)]
e4 = [randint(50, 100) for _ in xrange(10)]

count = 0

for s in chain(e1, e2, e3, e4):
    if s > 90:
        count += 1

print count

其运行结果为:

8

推荐阅读更多精彩内容