有没有一种类似python的方法来重写下面的代码块？列出理解？

Question

我正在试着做this。 这就是我想到的：

X, Y, A = [x1], [y1], [(x1 + y1) % F]
for i in range(1, N):
  X.append((C * X[i-1] + D * Y[i-1] + E1) % F)
  Y.append((D * X[i-1] + C * Y[i-1] + E2) % F)
  A.append((X[i] + Y[i]) % F)

这项工作，但正如你看到的，它不是很Python。我想知道是否有一种方法可以通过pythonic的方式获得相同的结果，可能是通过使用列表理解。 谢谢

Answer 1

如果x和y不依赖于A，我可能会尝试为它们创建一个生成器，然后从中构建X、Y和A（或者做任何事情）

def genXYA(n):
    x, y = x1, y1
    for i in range(n):
        nx, ny = (C * x + D * y + E1) % F, (D * x + C * y + E2) % F
        a = (nx + ny) % F
        yield nx, ny, a
        x, y = nx, ny


X, Y, A = map(list, zip(*genXYA(N)))

---

只是为了解释这里发生了什么

生成器就像一个列表，除了不立即返回所有内容外，它一次只返回一项

函数genXYA(n)返回一个生成器，该生成器在每一步返回元组(x, y, a)，并在返回n后停止

现在我们有了这个生成器，我们可以直接创建列表X、Y和A

说N = 5，如果你做了list(genXYA(N))，你会得到

[(x2, y2, a1),
 (x3, y3, a2),
 (x4, y4, a3),
 (x5, y5, a4),
 (x6, y6, a5)]

您想要的是将这些列分隔为单独的命名列表。您可以使用zip进行此操作zip获取了许多iterables，并创建了新的iterables，这些iterables是每个iterables的第一个、第二个、第三个，等等

为了给zip提供正确的参数，我们需要解包生成器。这就是*的目的

zip(*genXYA(N))的结果是另一个生成器，但是我们可以通过调用list(zip(*genXYA(N)))来查看其中的内容

[(x2, x3, x4, x5, x6), # => X
 (y2, y3, y4, y5, y6), # => Y
 (a1, a2, a3, a4, a5)] # => A

所以我们离最终结果越来越近了

当我们调用map(list, zip(*genXYA(N)))时，我们是说对于该列表中的所有内容，我们需要每个项目的列表（元组列表是包含元组中所有内容的列表）

map还返回一个生成器，然后我们可以将其解压到所需的变量中

---

使用生成器的好处是，只在需要时计算下一次迭代，而不必存储所有以前的迭代

例如，如果您只需要A的最后一个值，可以执行以下操作：

for x, y, a in genXYA(N):
    pass
print(a)