Eu tenho uma matriz que se parece com isso

[1, 0, 1 , 0 , 0, 1]

E eu quero obter os índices que têm 1 nele. Então, aqui eu teria uma variedade de[0, 2 , 5] e, com base nisso, eu criaria uma nova matriz que pegaria esses números e exponharia 2 com eles. Portanto, a matriz final é

[2**0, 2**2, 2**5]

Existe uma maneira de escrevê-lo o mais rápido possível?