numpyでブール条件のエンドポイントを見つけるより高速な方法がありますか？

Question

アレイのブール値プロパティの端点を見つけるための（より一般的な）高速な方法よりも知っている人はいませんか？

numpy.nonzero（a）[0] [ - 1]は、（dimension = 0）の最後の非ゼロ要素のインデックスであり、同様にnumpy.nonzero（a）[0] [0] is is最初の非ゼロ要素のインデックス

最初の要素または最後の要素のみを気にしていることが分かっている場合は、上記のように "nonzero"を実行するよりもメモリを少なくし、共通ケースの実行時間を短縮できます。たとえば、線形検索を使用する場合は、少なくとも適切な端で開始することができます（条件に一致する最後の値を見つけるために後方を検索する）。または、バイナリ検索を使用することもできます（たとえば、中間の要素が条件に一致する場合は、最初の要素が真である最後の要素を検索する必要はありません）。 これは、既存の実装が存在する可能性があると思われますが、そのようなものは見つかりませんでした。

Answer 1

ブール値配列の最初のTrue要素は、argmaxを使用して検索できます。

a = np.array([False, False, True, True, True]) 
first_True = a.argmax() 
last_True = len(a) - 1 - a[::-1].argmax() 

あなたはFalseの値を見つけるためにargminを使用することができ、これは速くなるとゼロ以外の使用するよりも少ないメモリがかかりますが、これはaの長さが直線的です。線形より高速にしたい場合は、aが "ソート"されていることを知る必要があります。ブール値の配列の場合は、ブロックがFalseで、その後にすべてTrueが続くことを意味します。その場合、ソートされた検索を使ってFalseとTrueの境界を見つけることができます。

first_True = a.searchsorted(True, 'left')