Индексирование массивов с разным количеством измерений

Question

Я использую numpy и хочу сохранить результат вычисления в четырехмерной матрице или . Я хочу сделать что-то вроде этого:

result[:, indices, iteration] = some_function()  # won't work

, где размерность понимается по длине кортежа indices. Так что если indices равно, скажем, (6,) или (8, 5), тогда я хочу получить эквивалент

result[:, 6, iteration] = some_function()

или

result[:, 8, 5, iteration] = some_function()

Обычно я помещаю все координаты в кортеж и использую это как индекс, но наличие : не позволяет мне сделать это.

Измерение result было ранее установлено на правильное измерение, поэтому можете проверить его, но было бы неплохо использовать только длину indices для его определения. Есть ли способ сделать это?

Answer 1

Попробуйте использовать result[(slice(None),) + indices + (iteration,)]

Answer 2

Если вам нужно знать длину индексов:

len(indices)

Но разве это ваш вопрос?