我有一个函数,我通过迭代 for 循环来填充一个 100x100 的二维数组。我需要将值存储在 [i][i]、[i][i-1] 和 [i][i+1] 位置。这是我目前所拥有的:
Delta = 1/100
N = 100
Matrix = np.zeros(N,N), int)
for i in range(N):
Matrix[i][i] = ((2*D)/pow(Delta,2) + SigmaA)
Matrix[i][i-1] = -(D/pow(Delta,2))
Matrix[i][i+1] = -(D/pow(Delta,2))
皈依舞
相关分类