Narray de indexação do número variável de dimensões
Eu tenho uma instância de numpy ndarray, mas de um tamanho variável.
import numpy as np
dimensions = (4, 4, 4)
myarray = np.zeros(shape = dimensions)
Nesse caso, obtenho uma forma "cúbica" da matriz e se eu quiser indexar uma fatia demyarray
Eu posso usarmyarray[:][:][0]
porque eu sei que existem 3 dimensões (eu uso 3 pares de[]
)
No caso de 4 dimensões, eu usariamyarray[:][:][:][0]
. Mas como o número de dimensões pode mudar, não posso codificá-lo dessa maneira.
Como posso indexar uma fatia dessa matriz, dependendo do número de dimensões? Parece um problema simples, mas não consigo pensar em nenhuma solução.