我正在寻找最具惯用性和高效性的方法来切割维数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数组中。
我正在寻找快速解决方案,因为阵列非常大。
谢谢!