将numpy数组从值转换为索引

Mic*_*ers 2 python arrays numpy

我有一个numpy数组a:[True,False,True,False,False,...]我喜欢有一个numpy数组,其索引为True和False值,即[0,2,... ]和[1,3,4,......]

Sve*_*ach 9

要获取True值的索引a,您可以使用

a.nonzero()
Run Code Online (Sandbox Code Playgroud)

对于False值的索引,请使用

(~a).nonzero()
Run Code Online (Sandbox Code Playgroud)