如何有效地計算numpy中的距離到最近的1個掩碼？

Question

在numpy中我有一個1和0的2d數組。我需要計算一個新的數組（相同的尺寸），其中每個元素包含距離掩模數組中對應點最近的1的距離。

例如

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

我需要b鍵看起來像這樣：

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

PS。我會在非常大的數組上做這個，所以效率越高越好！ 謝謝！

Answer 1

你要找的MATLAB的bwdist相當於;請查看this SO question瞭解更多詳情。簡短的答案是使用scipy.ndimage.morphology.distance_transform_edt。