Конвертировать массив numy из значений в индексы

Question

У меня есть массив значений: [True, False, True, False, False, ...] И мне нравится иметь массив NumPy, который имеет индексы значений True и False, то есть [0, 2, ...] и [1, 3, 4, ...]

Answer 1

Чтобы получить индексы значений True в a, вы можете использовать

a.nonzero()

Для индексов значений False используйте

(~a).nonzero()