特定の値のピクセルにメジアンフィルタを適用しますか？

Question

I内部にintを持つ行列があります。

隣人のいずれかが111である場合は、111をその直近の4近傍の中央値に置き換える必要がありますが、無視されます。例えばのために

： -

matrix = 1 2 3 4 5 6 
     101 111 1 3 44 3 
     111 3 4 4 5 6 
      1 2 4 5 7 7 

after replacing expected op 
1 2 3 4 5 6 
101 2.5 1 3 44 3 
3 3 4 4 5 6 
1 2 4 5 7 7 

私のコードはかなり悪いと、おそらく非常に遅いです。すべてのヘルプは

def median_fil_mat(matrix): 
    rows,columns= np.where(matrix==111) 
    r,c=np.shape(matrix) 
    for each_row in rows: 
     for each_colmn in columns: 
      if each_row==r-1: 
       r1=[each_row-1] 
      elif each_row>0 & each_row!=r-1: 
       r1= [each_row-1,each_row+1] 
      else: 
       r1=[each_row+1] 

      if each_colmn ==c-1: 
       c1=[each_colmn-1] 
      elif each_colmn >0 & each_colmn!=c-1: 
       c1=[each_colmn-1,each_colmn+1] 
      else: 
       c1=[each_colmn+1] 
      med_lis=list() 
      for rr in r1: 
       for cc in c1: 
        med_lis.append(matrix[rr,cc]) 
      med_lis=[x for x in med_lis if x!=111 ] 
      matrix[each_row,each_colmn]= np.median(med_lis) 
    return matrix 

Answer 1

def func(array): 
    if array[2]==111: 
     return np.median(array[array!=111]) 
    else: 
     return array[2] 
fp = np.array([[0, 1, 0], [1, 1, 1], [0, 1, 0]]) 
a = np.fromstring("""1 2 3 4 5 6 
     101 111 1 3 44 3 
     111 3 4 4 5 6 
      1 2 4 5 7 7""", sep=" ").reshape((4, 6)) 
generic_filter(a, func, footprint=fp, mode='nearest') 

戻り

array([[ 1. , 2. , 3. , 4. , 5. , 6. ], 
     [ 101. , 2.5, 1. , 3. , 44. , 3. ], 
     [ 3. , 3. , 4. , 4. , 5. , 6. ], 
     [ 1. , 2. , 4. , 5. , 7. , 7. ]])