Учитывая массив объектов:
a = numpy.array([[True,False],[True,True],[True,False,True]])
Ожидаемый результат:
[[0],[0,1],[0,2]]
Есть ли способ получить этот вывод, используя numpy .where () (может быть с numpy .vstack () или numpy .hstack ()) или некоторые другие функции numpy, о которых я не знаю?
Пока я пробовал:
c = [numpy.hstack(numpy.where(a[i])) for i in range(len(a))]
Это решает мою цель, но я хочу векторизовать ее, используя numpy функциональные возможности!
PS: Не нужно никаких циклов!