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 

但我也希望它生成ab 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 

有没有更优雅的方法来使用单个发电机获得这个 output？ 就像是：

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)将其扩展到无限的值序列。

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