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

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

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

1 个答案: