Pobranie bieżącego elementu w pliku scipy.ndimage.filters.generic_filter

Question

Czy możliwe jest uzyskanie "bieżącego elementu" w funkcji filtru scipy.ndimage.filters.generic_filter?

Jeśli, na przykład, A[0] zawsze zawierały bieżący element (co nie wydaje się być przypadek) coś jak następuje może znaleźć lokalne maksima

def local_max_f(A) : 
    return A[0] == A.max() 

img = np.random.rand(100).reshape(10,10) 
ndimage.generic_filter(img, local_max_f, size=3) 

Answer 1

Bieżący element powinien być elementem centrum size (lubw twoim przykładzie), ale nie można polegać na krawędziach tablicy wejściowej i zależy to od mode w zakresie obsługi granic tablic.

Parametr docs instead provide a neat example (dostosowany do poniższej sytuacji) do określania aktualnej pozycji w filtrze za pomocą klasy w celu zachowania pewnego stanu między kolejnymi iteracjami. To zawsze najpierw przechodzi przez ostatni wymiar; wszystko, co powinieneś zrobić, to zastąpić linię result dowolną funkcją filtrującą, w twoim przypadku będzie to coś w rodzaju result = self._array[self.coordinates] == buffer.max().

a = arange(12).reshape(3,4) 

class fnc_class: 
    def __init__(self, _array): 
     # store the shape: 
     self.shape = _array.shape 
     self._array = _array 
     # initialize the coordinates: 
     self.coordinates = [0] * len(self.shape) 

    def filter(self, buffer): 
     result = self._array[tuple(self.coordinates)] == buffer.max() 
     print self.coordinates 
     # calculate the next coordinates: 
     axes = range(len(self.shape)) 
     axes.reverse() 
     for jj in axes: 
      if self.coordinates[jj] < self.shape[jj] - 1: 
       self.coordinates[jj] += 1 
       break 
      else: 
       self.coordinates[jj] = 0 
     return result 

fnc = fnc_class(a) 
generic_filter(a, fnc.filter, footprint = [[1, 0], [0, 1]])