將numpy數組從數值轉換爲索引

Question

我有一個numpy數組a：[True，False，True，False，False，...] 我喜歡有一個numpy數組，其索引爲True和False值，即[0,2，...]和[1，3，4，...]

Answer 1

爲了得到True值的索引中a，您可以使用

a.nonzero() 

對於False值的索引，使用

(~a).nonzero()