遍历两个列表

Question

我有两个列表list1和list2数字，我想使用相同的指令遍历它们。像这样：

for item in list1:
  print(item.amount)
  print(item.total_amount)

for item in list2:
  print(item.amount)
  print(item.total_amount)

但这感觉多余。我知道我会写for item in list1 + list2:，但是要付出一定的时间。

有没有办法做到这一点而又不浪费时间呢？

Answer 1

这可以通过以下方式完成
itertools.chain
：

import itertools

l1 = [1, 2, 3, 4]
l2 = [5, 6, 7, 8]

for i in itertools.chain(l1, l2):
    print(i, end=" ")

将打印：

1 2 3 4 5 6 7 8

根据文档，请chain执行以下操作：

创建一个迭代器，该迭代器从第一个可迭代对象返回元素，直到耗尽为止，然后继续进行下一个可迭代对象，直到所有可迭代对象都耗尽为止。

如果列表中有列表，
itertools.chain.from_iterable
则可用：

l = [l1, l2]
for i in itertools.chain.from_iterable(l):
    print(i, end=" ")

产生相同的结果。

如果您不想为此导入模块，则为其编写函数非常简单：

def custom_chain(*it):
    for iterab in it:
        yield from iterab

这需要Python 3（对于Python 2），只需yield使用循环即可：

def custom_chain(*it):
    for iterab in it:
        for val in iterab:
            yield val

除了以前的版本外，Python还3.5具有扩展的拆包概述，还允许在列表文字中拆包：

for i in [*l1, *l2]:
    print(i, end=" ")

尽管这比l1 + l2它仍然构造一个列表然后将其扔掉要快一些；仅将其作为最终解决方案。