将numpy数组从值转换为索引

时间:2012-02-15 16:10:19

标签: python arrays numpy

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

1 个答案:

答案 0 :(得分:9)

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

a.nonzero()

对于False值的索引,请使用

(~a).nonzero()