如何从python中的两个列表生成一个生成器？

Question

给出列表a,b

a = [5, 8, 9]
b = [6, 1, 0]

我想创建一个生成器gen,以便:

for x in gen:
  print x

输出

5, 8, 9, 6, 1, 0

Answer 1

你可以使用itertools.chain:

>>> from itertools import chain
>>> a = [5, 8, 9]
>>> b = [6, 1, 0]
>>> it=chain(a,b)
>>> for x in it:
    print x,
...     
5 8 9 6 1 0