找到值为true的布尔数组的索引

时间:2016-04-29 14:44:37

标签: python python-3.x numpy boolean

我想知道找到Boolean数组的所有索引的最佳方法,其值为True。例如,bool值数组

import numpy as np
A = np.array([True, False, True, False, True])
true_list = A[A == True].index.tolist()

1 个答案:

答案 0 :(得分:6)

这样做很容易:

np.where(A)

或者:

A.nonzero()