2

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

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
4

3 回答 3

4

你根本不需要循环。假设v和是带有条目的一维数组，您可以vnew这样unchangeableflag做size*size

v = v.reshape(size, size)
vnew = vnew.reshape(size, size)
unchangeableflag = unchangeableflag.reshape(size, size)
average = v[1:-1, 2:]
average += v[1:-1, :-2] 
average += v[2:, 1:-1]
average += v[-2:, 1:-1]
average /= 4.0
vnew[1:-1, 1:-1][unchangeableflag[1:-1, 1:-1] == 0] = average

但是你实际上想要达到什么目的？这看起来很可疑，就像你可以通过离散拉普拉斯算子的一​​些应用逃脱。

（请注意，这里假设v包含浮点数。如果dtype`v' 是 sime 整数类型，则需要稍作修改。）

于 2011-01-28T15:31:28.653 回答
3

您应该考虑使用 SciPy 的卷积过滤器或generic_filter。这仍然是计算密集型的，但比循环快得多。通常，在进行这种类型的平均时，中心元素也包括在内。请注意，这些解决方案也适用于多维数组。

from scipy import ndimage
footprint = scipy.array([[0,0.25,0],[0.25,0,0.25],[0,0.25,0]])
filtered_array = scipy.convolve(array, footprint)

或者

from scipy import ndimage
def myfunction(window):
    return (window[0,1] + window[1,0] + window[1,2] + window[2,1]) / 4
filtered_array = scipy.generic_filter(array, myfunction, size=3)
于 2011-01-28T15:44:35.340 回答
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
于 2011-01-28T15:17:06.230 回答