効率的にnumpyのマスクで1に近い距離を計算するには？

Question

numpyでは、私は1と0の2次元配列を持っています。新しい配列（同じ次元）を計算する必要があります。各要素には、マスク配列内の対応する点から最も近い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です。