Tengo una instancia de ndarray numpy, pero de tamaño variable.

import numpy as np
dimensions = (4, 4, 4)
myarray = np.zeros(shape = dimensions)

En este caso, obtengo una forma "cúbica" de la matriz y si quiero indexar una porción demyarray Puedo usarmyarray[:][:][0] porque sé que hay 3 dimensiones (uso 3 pares de[])

En caso de 4 dimensiones, usaríamyarray[:][:][:][0]. Pero dado que el número de dimensiones puede cambiar, no puedo codificarlo de esta manera.

¿Cómo puedo indexar un segmento de dicha matriz en función del número de dimensiones? Parece un problema simple, aunque no se me ocurre ninguna solución.