如何從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

你可以使用generator expressions一個可笑的Python的優雅的一行：

>>> 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