查找其值为true的布尔数组的索引

Question

I am wondering whats the best way to find all the indices of a Boolean array, of which the values are True . For example, an array of bool values,

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

Answer 1

This will do it easily:

np.where(A)

Or:

A.nonzero()