python - 使用 numpy.where() 进行矢量化

Question

给定一个对象数组：

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：不想要任何循环！