argsort の動作が謎だ（謎ではありませんでした）

Question

謎なこと

np.argsortを使うと，配列のソート後のインデックスの配列を作ってくれるということなのですが，期待した動作になりません。xの[2, 0, 1]の行はソートしたら[0, 1, 2]だから，out[7]の[1, 2, 0]ではなく，[2, 0, 1]が正しい結果になるべきではないかと思うのですが，謎です。この関数の使い方を教えてください。

In [6]: x                                                              
Out[6]: 
array([[2, 0, 1],
       [4, 3, 5],
       [6, 8, 7]])
In [7]: np.argsort(x)                                                  
Out[7]: 
array([[1, 2, 0],
       [1, 0, 2],
       [0, 2, 1]])

Answer 1

あなたの考え

value	index	結果
2	index=0	→	new index=2
0	index=1	→	new index=0
1	index=2	→	new index=1

numpy.argsort()の動作

value	index	結果
2	index=0	→	old index=1
0	index=1	→	old index=2
1	index=2	→	old index=0

Comments

[jbe00214]

ありがとうございます。どうやら新しいインデックスで考えていたようです。古いインデックスの配列を作るのですね。

Answer 2

xの[2, 0, 1]の行はソートしたら[0, 1, 2]だから

はい。元の位置でいえば1番目の値の0、2番目の位置の値の1、0番目の値の2の順番となります。numpy.argsortはソートされた値の元の要素位置を返すので[1,2,0]で正しいです。

Comments

[jbe00214]

ソートされた後の値の，元の位置の配列なんですね。何か勘違いをしていたのかもしれません。ありがとうございました。