python - 迭代 numpy 数组的函数

Question

我编写了一个简单的函数，它使用一些固定索引迭代 numpy 数组。

def compute_V(i,j,nA, nB,V):
    Vijkl = np.zeros((i,j,nA,nB))
    for k in range(nA):
        for l in range(nB):
            Vijkl[i,j,k,l] = V[i,j,k,l] + 3

    return Vijkl

我收到以下错误: IndexError:索引 1 超出尺寸为 1 的轴 0 的范围

我做错了什么？

矩阵 V 的形状为:(1, 2, 1, 2) 并且看起来:

[[[[-0.00009 -0.00001]]

  [[-0.00001 -0.00001]]]] 

i = 0, j = 0, nA = 1, nB = 2 

看起来循环进行了一次迭代:

i,j,k,l,V:  0 0 0 0 -9.39073120245e-05

然后抛出错误。

Answer 1

Vijkl 的形状为 (i, j, nA, nB)，因此每个维度的最大限制为 (i-1, j-1, nA-1, nB-1)。您超出了前两个维度的最大范围。