numpy画像から色付きの線を抽出します。

Question

色付きの曲線を持つ画像があります。

各行ごとに曲線の上位ピクセルのインデックスを抽出します。

このコードは動作しますが、（fがmagicは、線の色のRGBコードで、イメージです）かなり遅いです：

res = f.shape[0] 
magic = 146, 47, 6 
return [min(l for l in range(res) if np.array_equal(magic, f[l, c, :3])) for c in range(res)] 

にはどうすれば速く同じことを行うことができますか？

Answer 1

magicの要素を各行、つまり最後の軸に沿って比較します。これによりマスク2Dが得られます。

mask = (f == magic).all(-1) 

以来、我々は最終的にそのマスク配列の最初のTrueのインデックスを見つけることを意味するであろうと、min(l)を行っています。それは動作しませんので、スカラーを返しARGMAX

(f == magic).all(-1).argmax(0)