a、b、aa、ab、ba、bb、aaa、aab 的生成器

Question

这个：

import itertools

gen1 = itertools.product('ab', repeat=3)
for i in gen1:
    print(''.join(i), end=' ')

打印这个：

aaa aab aba abb baa bab bba bbb 

但我也希望它生成a b aa ab ba bb。我可以这样做：

import itertools

for i in range(1, 4):
    gen2 = itertools.product('ab', repeat=i)
    for j in gen2:
        print(''.join(j), end=' ')

打印这个：

a b aa ab ba bb aaa aab aba abb baa bab bba bbb 

有没有更优雅的方法来使用单个生成器获取此输出？比如：

import itertools

gen3 = (your code here)
for i in gen3:
    # prints forever in the pattern: a b aa ab ba bb aaa aab aba abb baa bab bba bbb
    print(''.join(i), end=' ')

Answer 1

我会将多个迭代器链接在一起：

from itertools import chain, product


gens = chain.from_iterable(product('ab', repeat=i) for i in range(1,4))
for i in gens:
  print(''.join(i), end=' ')

如果需要，可以通过将range(1,4) 替换为itertools.count(1)，将其扩展到无限的值序列。

如果您的“永远打印”注释意味着您想在 bbb 之后返回到 a，则将链式迭代器包装在 itertools.cycle 中。