python - 如何从 python 中的两个列表创建生成器？

Question

给出列表 a, b

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

我想创建一个生成器，这样：

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

Answer 2

def chain(*args):
    for arg in args:
        for item in arg:
            yield item

a = [5, 8, 9]
b = [6, 1, 0]
for x in chain(a,b):
    print x,

print ', '.join(map(str,chain(a,b)))

Answer 3

您可以将生成器表达式用于可笑的 Pythonic 和优雅的单线：

>>> a=[5,8,9]
>>> b=[6,1,0]
>>> g=(i for i in a+b)

测试：

>>> for i in g:
       print i

5
8
9
6
1
0

或测试 #2，如果您真的更喜欢在每个项目之间使用逗号，：

>>> print ', '.join(map(str,g))

5, 8, 9, 6, 1, 0