Индексирование ndarray переменного числа измерений
У меня есть экземпляр numpy ndarray, но с переменным размером.
import numpy as np
dimensions = (4, 4, 4)
myarray = np.zeros(shape = dimensions)
В этом случае я получаю «кубическую» форму массива, и если я хочу проиндексировать фрагментmyarray
я могу использоватьmyarray[:][:][0]
потому что я знаю, что есть 3 измерения (я использую 3 пары[]
).
В случае 4-х размеров, я бы использовалmyarray[:][:][:][0]
, Но так как число измерений может измениться, я не могу жестко закодировать это таким образом.
Как я могу индексировать срез такого массива в зависимости от количества измерений? Похоже, простая проблема, не могу придумать никакого решения, хотя.