Python 如何优化numpy均值窗口

Python 如何优化numpy均值窗口,python,numpy,gdal,Python,Numpy,Gdal,我正试图得到一个新的光栅,使其在所有阵列中循环,并使用一个25x25像素的搜索窗口。我想知道你是否知道一个更好的方法,因为我的方法花费了太多的时间 import sys import os import numpy as np import math from osgeo import gdal, osr, gdal_array, gdalnumeric from osgeo.gdalnumeric import * numpy.warnings.filterwarnings('ignore

我正试图得到一个新的光栅,使其在所有阵列中循环,并使用一个25x25像素的搜索窗口。我想知道你是否知道一个更好的方法,因为我的方法花费了太多的时间

import sys
import os
import numpy as np
import math

from osgeo import gdal, osr, gdal_array, gdalnumeric
from osgeo.gdalnumeric import *

numpy.warnings.filterwarnings('ignore')

def mean_neighbors(M,x,y,w=1):
    l = []
    for i in range(max(0,x-w), x+(w+1)):
        for j in range(max(0,y-w), y+(w+1)):
            try:
                t = M[i][j]
                l.append(t)
            except IndexError:
                pass
    return np.mean(l)

raster_file = gdal.Open('image.tif', gdal.GA_ReadOnly)
rst         = gdalnumeric.BandReadAsArray(raster_file.GetRasterBand(1))

cob = np.zeros(rst.shape)

for i in range(rst.shape[0]):
    for j in range(rst.shape[1]):
        cob[i][j] = mean_neighbors(rst, i, j, 25) # want to optimize this function

移动窗口是一个非常常见的函数,您不需要自己编写。以下是两种可以使用的快速实现:

  • 移动是指

如果需要,您可以尝试列表理解