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) 提取一个特定的高度。此信息存储在二维数组中。


我正在寻找一个快速的解决方案,因为数组非常大。

谢谢!

4

0 回答 0