Я нашел несколько вопросов, которые близки к тому, что я задаю, но они настолько различны, что, кажется, не решают мою проблему. Я пытаюсь взять 1d срез вдоль одной оси для ndarray. В качестве примера для трехмерного массива
[[[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8]],
[[ 9,10,11],
[12,13,14],
[15,16,17]],
[[18,19,20],
[21,22,23],
[24,25,26]]]
Я хочу следующие 1d ломтики
[0,1,2]
...
[24,25,26]
[0,3,6]
...
[20,23,26]
[0,9,18]
...
[8,17,26]
, что эффективно приравнивается к следующему (для трехмерных массивов):
ary[i,j,:]
ary[i,:,k]
ary[:,j,k]
Я хочу обобщить это в массив из n измерений
(для двумерного массива мы бы получили ary [i ,:] и ary [:, j] и т. Д.)
Есть ли функция numpy, которая позволяет мне делать это?
РЕДАКТИРОВАТЬ: Исправлена индексация 2-го измерения