Mic*_*ers 2 python arrays numpy
我有一个numpy数组a:[True,False,True,False,False,...]我喜欢有一个numpy数组,其索引为True和False值,即[0,2,... ]和[1,3,4,......]
要获取True值的索引a,您可以使用
a.nonzero()
Run Code Online (Sandbox Code Playgroud)
对于False值的索引,请使用
(~a).nonzero()
Run Code Online (Sandbox Code Playgroud)