Нарезанный массив N измерения с массивом измерения N-1
Я ищу самый идиоматичный и эффективный способ нарезать массив измерения N с массивом измерения N-1, который представляет индекс последнего измерения нарезать. Выход будет иметь размерность N-1.
В качестве примера:
import numpy as np
array = np.array([[[1,2], [3,4]], [[5,6], [7,8]]])
index = np.array([[0, 1], [1, 0]])
Результаты будут:
np.array([[1, 4], [6, 7]])
Например, можно представить массив (lat, lon, alt), и мы хотим извлечь конкретную высоту для каждого (lat, lon). Эта информация хранится в 2D массиве.
Я ищу быстрое решение, так как массивы довольно большие.
Спасибо!