Question

我编写了一个生成函数，它以与Excel等电子表格应用程序中的列命名方案相同的方式产生无限的字符串序列，例如：

'', 'A', 'B', ... 'Z', 'AA', 'AB', ... 'AZ', 'BA', ... 'ZZ', 'AAA', ...

我的功能没有任何问题：

def __suffix_generator():
    len, gen = (0, iter(['']))

    while True:
        try:
            suffix = next(gen)
        except StopIteration:
            len += 1
            gen = itertools.product(string.ascii_uppercase, repeat=len)
            suffix = next(gen)

        yield ''.join(suffix)

但是，我想把它变成一个更惯用的版本，它只使用生成器表达式，到目前为止我的最佳镜头是：

def __suffix_generator_gexp():
    from itertools import product, count
    from string import ascii_uppercase

    return (''.join(suffix) for suffix in
        (product(ascii_uppercase, repeat=len) for len in count()))

使用该生成器时，我得到一个运行时TypeError，它告诉我suffix变量的类型不被接受：

TypeError: sequence item 0: expected string, tuple found

我的假设是suffix应该是包含特定组合字母的元组，并且join会将其转换为字符串。我怎样才能让它正常工作，就像第一个函数一样？

Answer 1

也许这就是你要找的东西：

map(''.join, chain.from_iterable(product(ascii_uppercase, repeat=l) for l in count(1)))

count(n)使用n生成以step = 1开头的数字序列，以便每个数字N_{t+1} = N_t + step和N_1 = n。

如果您正在使用Python 2.x，map将尝试构建列表并将失败，因此您可以这样做：

(''.join(perm) for perm in (...))

...是第一个代码段中map的第二个参数。

如何将此生成器函数转换为生成器表达式？

1 个答案: