Jak skutecznie obliczyć odległość do najbliższej 1 w masce w numpy?

Question

W numpy mam tablicę 2d z 1s i 0s. Muszę obliczyć nową tablicę (te same wymiary), gdzie każdy element zawiera odległość do najbliższego 1 od odpowiedniego punktu w tablicy masek.

np.

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

muszę b, aby wyglądać tak:

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

PS. Będę to robić na bardzo dużych tablicach, więc im bardziej wydajny, tym lepiej! Dzięki!

Answer 1

Poszukujesz odpowiednika MATLAB-a bwdist; sprawdź numer this SO question, aby uzyskać więcej informacji. Krótka odpowiedź to użycie scipy.ndimage.morphology.distance_transform_edt.