如何有效地計算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 ; 查看此SO問題了解更多詳情。 簡短的回答是使用scipy.ndimage.morphology.distance_transform_edt 。