python生成具有特定条件的无限列表

Question

我知道python中有生成器yield，例如：

def f(n): 
    x = n
    while True:
        yield x
        x = x+1

所以我尝试在不使用 iterate 的情况下将此 haskell 函数转换为 python：Haskell infinite recursion in list comprehension

我不确定如何在 python 中定义基本情况，也不确定如何将 if 语句与这个 yield 工作人员结合起来！这是我尝试做的：

def orbit(x,y):
    while True:
        yield p (u,v)
        p (u,v) = (u^2 - v^2 + x, 2 * u * v + y)

Answer 1

我看不出你从哪里得到p。据我所知，您几乎可以从 Haskell 中直接翻译它：

def orbit(x, y):
    u, v = 0, 0
    while True:
        u, v = u**2 − v**2 + x, 2*u*v + y
        yield u, v

在他们的示例中，将函数调用为 orbit(1, 2)、u 将在第一轮中绑定到 1 和 v 到 2，然后生成 ((1, 2))。在下一次迭代中，u = 1**2 - 2**2 + 1 = 1 - 4 + 1 = -2 和 v = 2*1*2 + 2 = 6。