Numpy фрагмент произвольных размеров ⇐ Python

[Anonymous]

Я хотел бы разрезать массив numpy, чтобы получить i-й индекс в последнем измерении. Для 3D-массива это будет:

Код: Выделить всё

slice = myarray[:, :, i]

Но я пишу функцию, в которой я могу взять массив произвольных размеров, поэтому для 4D-массива мне понадобится myarray[:, :, :, i] , и так далее. Есть ли способ получить этот фрагмент для любого массива без необходимости явно записывать размеры массива?

Подробнее здесь: https://stackoverflow.com/questions/121 ... dimensions