У меня есть экземпляр numpy ndarray, но с переменным размером.

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

В этом случае я получаю «кубическую» форму массива, и если я хочу проиндексировать фрагментmyarray я могу использоватьmyarray[:][:][0] потому что я знаю, что есть 3 измерения (я использую 3 пары[]).

В случае 4-х размеров, я бы использовалmyarray[:][:][:][0], Но так как число измерений может измениться, я не могу жестко закодировать это таким образом.

Как я могу индексировать срез такого массива в зависимости от количества измерений? Похоже, простая проблема, не могу придумать никакого решения, хотя.