numpy数组-从原始数组和具有索引值的数组创建新数组

Question

我有两个numpy和A数组。

B的形状为（200，80,000），是稀疏向量

A的形状为（200，3），由B派生而来；它代表了200个“行”中每行中最大的3个值的3个索引“列”位置，从而使

A

我想创建一个向量B = np.argsort(-A, axis=1)[:,1:4] ，其形状为（200，3），它根据索引位置指示从C返回的 value A。

我最初尝试过

B

但是创建了一个（200，200，3）矩阵。基本上，我想从A[:, B] 获取200个数组中每个数组的三个列索引号，并从B中提取值。我已经附上了一个“玩具”示例的图片，以演示我希望做的事情。我将不胜感激。