Pesquisando uma sequência em uma matriz NumPy
Digamos que eu tenho a seguinte matriz:
array([2, 0, 0, 1, 0, 1, 0, 0])
Como obtenho os índices em que tenho ocorrência de sequência de valores:[0,0]
? Portanto, a saída esperada para esse caso seria:[1,2,6,7]
.
Editar:
1) Observe que[0,0]
é apenas uma sequência. Poderia ser[0,0,0]
ou[4,6,8,9]
ou[5,2,0]
, apenas qualquer coisa.
2) Se minha matriz foi modificada para:array([2, 0, 0, 0, 0, 1, 0, 1, 0, 0])
, o resultado esperado com a mesma sequência de[0,0]
seria[1,2,3,4,8,9]
.
Estou procurando algum atalho NumPy.