itertools还是手写生成器，哪个更好？

Question

我有几个Python生成器，想把它们合并成一个新的生成器。我可以通过手动编写一个生成器，使用一堆yield语句来轻松实现这个目标。

另一方面，itertools模块就是为这种事情设计的，感觉用这个模块里的各种迭代器来创建我需要的生成器是更符合Python风格的做法。

不过，眼下这个问题变得相当复杂（这个生成器需要保持某种状态——比如正在处理的是第一个项目还是后面的项目——第i个输出还依赖于第i个输入项目的条件，而各种输入列表在合并到生成的列表之前需要以不同的方式处理）。

由于标准迭代器的组合几乎让人难以理解（因为写代码是线性的），我在想使用标准的itertools生成器和手动编写的生成器函数相比，是否有什么好处（无论是基础情况还是更复杂的情况）。实际上，我觉得在90%的情况下，手动编写的版本更容易阅读——可能是因为它们的写法更直接，而不是像链式迭代器那样的函数式风格。

编辑

为了说明我的问题，这里有一个（简单的）例子：设a和b是两个相同长度的可迭代对象（输入数据）。a里的项目是整数，b里的项目本身也是可迭代的，每个项目是字符串。输出应该对应于以下生成器函数的输出：

from itertools import *
def generator(a, b):
    first = True
    for i, s in izip(a, b):
        if first:
            yield "First line"
            first = False
        else:
            yield "Some later line"
        if i == 0:
            yield "The parameter vanishes."
        else:
            yield "The parameter is:"
            yield i
        yield "The strings are:"
        comma = False
        for t in s:
            if comma:
                yield ','
            else:
                comma = True
            yield t

如果我用函数式风格和itertools模块写同样的程序，最后得到的代码大概是这样的：

from itertools import *
def generator2(a, b):
    return (z for i, s, c in izip(a, b, count())
            for y in (("First line" if c == 0 else "Some later line",),
                      ("The parameter vanishes.",) if i == 0
                      else ("The parameter is:", i),
                      ("The strings are:",),
                      islice((x for t in s for x in (',', t)), 1, None))
            for z in y)

示例

>>> a = (1, 0, 2), ("ab", "cd", "ef")
>>> print([x for x in generator(a, b)])
['First line', 'The parameter is:', 1, 'The strings are:', 'a', ',', 'b', 'Some later line', 'The parameter vanishes.', 'The strings are:', 'c', ',', 'd', 'Some later line', 'The parameter is:', 2, 'The strings are:', 'e', ',', 'f']
>>> print([x for x in generator2(a, b)])
['First line', 'The parameter is:', 1, 'The strings are:', 'a', ',', 'b', 'Some later line', 'The parameter vanishes.', 'The strings are:', 'c', ',', 'd', 'Some later line', 'The parameter is:', 2, 'The strings are:', 'e', ',', 'f']

这可能比我最初的解决方案更优雅，但看起来像是一段写一次后就不理解的代码。我在想，这种写生成器的方式是否有足够的优势值得去这样做。

附言：我觉得我在函数式解决方案上的问题部分是因为为了减少Python中的关键字数量，一些关键字像“for”、“if”和“else”被重新利用在表达式中，这样它们在表达式中的位置就需要适应（生成器表达式z for x in a for y in x for z in y的顺序，至少对我来说，看起来不如经典的for循环顺序自然：for x in a: for y in x: for z in y: yield z）。

迭代器可迭代对象生成器编程风格代码可读性函数式编程状态管理组合模式

itertools还是手写生成器，哪个更好？

1 个回答

撰写回答