根据条件选择数组中的最大值

Question

我有一个numpy数组[6,5,4,3,2,1,0,1]来定义一个指标， 我根据a从这个数组中选择了一些索引 条件。数组索引为[1,2]。现在我想选择元素 在度量数组中具有相应最大值的所选索引中。

答案是1，因为所选索引中第1个元素是最大的。

Answer 1

我想你想要：

import numpy as np

x = np.array([6,5,4,3,2,1,0,1])
idx = np.array([1,2])

y = idx[ np.argmax(x[idx]) ]