如何为此添加一个循环？

Question

我的部分代码是：

list1 = zeros((x,y))

for j in range(1,y):
    for i in range(1, x-1):
        list1[i,j] = list1[i,j-1] + Equation

这很好用。但是，当我想进入下一阶段时，我需要修改第二个for循环中的“Equation”部分。假设等式为(a*b+c)*d，我希望parameters(a,b,c,d)中的一个随着j的每次增加而变化。

也就是说，当j为1时，a = something。当j增加到2时，会发生变化。它就像是j的功能。例如：a = A*cos(w*j)。

我的问题是，如何将此关系循环到代码中，以便每次都更新a？

Answer 1

只需在外部循环中添加表达式，根据a的变化值计算j：

for j in range(1, y):
    a = A * cos(w * j)
    for i in range(1, x-1):
        list1[i, j] = list1[i, j - 1] + (a * b + c) * d