python - 定義された軸で最大値のインデックスを使用

Question

行列「A」の各行の最大値の同じインデックスで配列「B」の値を抽出したいと考えています。インデックスを見つけるには、次のように numpy 関数「numpy.argmax」を使用します。

>>> A=numpy.asarray([[0,1,6],[3,2,4]]);A
array([[0, 1, 6],
       [3, 2, 4]])
>>> argA=numpy.argmax(A,axis=1);argA
array([2, 2])

問題は、「argA」を使用して配列「B」の値を抽出する方法がわからないことです

score 4 · Accepted Answer

の各エントリはargA、対応する行内の最大値のインデックス位置に対応します。行は ( を使用しているため) 明示的ではありませんaxis=1が、各エントリのインデックスに対応しています。したがって、目的の要素を取得するには、それらを追加する必要があります。

>>> A[[0,1], argA]
array([6, 4])

そう：

>>> B
array([[ 9,  8,  2],
       [ 3,  4,  5]])
>>> B[[0,1], argA] = 84,89
>>> B
array([[ 9,  8, 84],
       [ 3,  4, 89]])

使用を一般化するには：

>>> B[np.arange(A.shape[0]),argA]

score 0 · Accepted Answer

配列をインデックスとして直接使用できます。これは実際には次のように簡単です。

B[:, arga]

例えば：

>>> A[:,argA]
array([[6, 6],
       [4, 4]])

score 0 · Accepted Answer

これは少し非効率に見えるかもしれませんが、以下を使用できます。

A.take(argA, axis=1).diagonal()

代わりにA.take使用できますA[:, argA]がtake、軸についてより明示的です。

ravel_multi_indexフラットインデックスに変換するために使用できます：

A.take(np.ravel_multi_index((np.arange(len(argA)), argA), A.shape))

3 に答える 3