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

Question

我想知道什么是找到Boolean数组的所有索引的最佳方法,其中值是True.例如,一组bool值,

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

Answer 1

这样做很容易:

np.where(A)

要么:

A.nonzero()