python - 在 python 中为一组相似的语句创建一个循环

Question

我有一个名为 A0 的二维矩阵

A0 = [[0 for x in range(3)] for y in range(3)]

我有一个名为 newProb 的函数，它将其作为参数并按以下方式返回另一个 2D 矩阵:-

A1 = newProb(A0)

所以，我想将此代码放入循环中。

A1 = newProb(A0)
A2 = newProb(A1)
A3 = newProb(A2)
A4 = newProb(A3)

任何帮助将不胜感激。谢谢!

附注我必须以上述方式对该函数进行 100 次调用

Answer 1

与其尝试在循环中创建局部变量，不如将结果存储在列表中。您的列表索引将与您的命名约定很好地保持一致。

A = []

A.append([[0 for x in range(3)] for y in range(3)])

A[1] = newProb(A[0])
A[2] = newProb(A[1])
# ...

然后您可以轻松地将其放入带有范围的循环中。

# start the range off by 1
for i in range(1, 10):
    A[i] = newProb(A[i - 1])

这也可以写成

for i in range(10):
    A.append(newProb(A[i]))