Допустим, у меня есть многомерный массив numpy A. Я пытаюсь получить кусочки A в любом измерении, которое я выберу, например, если A - это трехмерный массив, и мне дают целое число layer и некоторые размерность dim:
Если dim = 2, выберите A [:,:, layer]
Если dim = 1, выберите A [:, layer,:]
Если dim = 0, выберите A [layer,:,:]
В моем случае у A может быть много измерений на выбор, поэтому я не хочу писать это вручную. Что такое простой, читабельный способ сделать это?