已经有Python生成器可以做这些基本的事情了吗？

Question

我发现自己使用这两个自定义生成器并思考“必须有一个 itertools 函数或已经做到这一点的东西！但没有找到任何东西。我错过了什么吗？谢谢！

def gothru(iters):
  for i in iters:
    for j in i:
      yield j

def overnover(fn,startval):
  val = startval
  while True:
    val = fn(val)
    yield val

编辑：我后来想象如何overnover 可用于生成斐波那契数列，我意识到需要对其进行泛化以允许该函数具有多个参数，

def overnover(fn,*args):
  while True:
    args = fn(*args)
    return args

然后您可以执行以下操作：

fibInfo = overnover(lambda x,y: (x+y, x), 1, 1)

-> (2,1) ... ( 3、 2) ... (5, 3) ... (8, 5) ... 然后：

fib = imap(lambda x:x[0], fibInfo)

-> 2 ... 3 ... 5 ... 8 ...

谢谢大家！

Answer 1

第一个是 chain.from_iterable。

与overnover最接近的是tabulate：

def tabulate(function, start=0):
    "Return function(0), function(1), ..."
    return imap(function, count(start))

这是函数的一种特殊情况，它输出连续的数字。 count 也需要一个 step，因此您可以将其概括为

def tabulate(function, start=0, step=1):
    "Return function(0), function(0+step), ..."
    return imap(function, count(start, step))

以下是 overnover 的一个版本，它可以让您将值发送到序列中：

def overnover(fn, val):
    while True:
        val = fn(val)
        val = (yield val) or val