条件で配列を最大に選択

Question

を定義するnumpy配列[6,5,4,3,2,1,0,1]があり、 条件に従ってこの配列からいくつかのインデックスを選択しました。配列インデックスは[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]) ]