я есть массив, который выглядит так

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

И я хочу получить те индексы, которые имеют 1 в нем. Так что здесь я бы получить массив[0, 2 , 5] а затем на основе этого я бы создал новый массив, который принимает эти числа и возводит в степень 2 с ними Таким образом, конечный массив

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

Есть ли способ написать это как можно быстрее?