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.