Numpe olarak maskeye en yakın 1 mesafe nasıl hesaplanır?

Question

Numpy'de 1d ve 0s 2d dizilim var. Her bir öğenin maske dizisindeki karşılık gelen noktadan en yakın 1'e olan mesafeyi içerdiği yeni bir dizi (aynı boyutlar) hesaplamam gerekiyor.

örn.

a=np.array(
[[1,1,0], 
[1,0,0], 
[1,0,0]]) 

Böyle bakmak b gerekir:

array([[0,0,1], 
     [0,1,1.41], 
     [0,1,2]]) 

PS. Bunu çok büyük diziler üzerinde yapacağım, bu yüzden daha verimli daha iyi! Teşekkürler!

Answer 1

MATLAB'ın bwdist; Daha fazla bilgi için this SO question'a bakın. Kısa cevap scipy.ndimage.morphology.distance_transform_edt kullanmaktır.