切片尺寸为N的numpy数组,其尺寸为N-1

时间:2018-04-09 12:56:54

标签: python numpy slice numpy-slicing

我正在寻找最具惯用性和高效性的方法来切割维数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数组中。

我正在寻找快速解决方案,因为阵列非常大。

谢谢!

0 个答案:

没有答案