python - Pythonで2つのリストからジェネレータを作成するにはどうすればよいですか?

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

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