加速一个计算大数组中相邻元素平均值的程序

Question

我对程序的速度有问题。我要计算一个大数组中四个邻居的平均数。这是我的一部分代码。你有什么办法改变最后一行吗？或者我应该使用另一个数组吗？在

for a in np.arange(100000):
    for x in np.arange(size):
        for y in np.arange(size):
            if unchangeableflag[x*size+y] == 0:
                vnew[x*size+y] = (v[(x+1)*size+y] + v[(x-1)*size+y] + v[x*size+y+1] + v[x*size+y-1]) / 4.0

Answer 1

我不确定，但你可以去掉一些不变量。在

for a in np.arange(100000):
    for x in np.arange(size):
        for y in np.arange(size):
            t = x*size+y
            if unchangeableflag[t] == 0:
                vnew[t] = (v[t+size] + v[t-size] + v[t+1] + v[t-1]) / 4.0