我正在寻找最惯用和最有效的方法来切片一个维度为 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) 提取一个特定的高度。此信息存储在二维数组中。
我正在寻找一个快速的解决方案,因为数组非常大。
谢谢!