Use argsort

tags: python Numpyパッケージにリストを並び替えるための argsort という関数があるが，その使い方がちょいと難しいからメモ．

>>> a= np.array([3,1,2])
>>> b= np.argsort(a)
>>> b
array([1, 2, 0])
>>> a
array([3, 1, 2])
>>> c= np.array(a)[b]
>>> a
array([3, 1, 2])
>>> b
array([1, 2, 0])
>>> c
array([1, 2, 3])

この例を見ると， a=[3,1,2] の argsort で得られたリスト b=[1,2,0] は， a をソートするとしたらこの順番ですよという a の要素番号のリストとなっている． なので， [a[b[0]],a[b[1]],a[b[2]]] というリストを作れば a を並び替えられることになる． これを実行しているのが，

c= np.array(a)[b]
# or
c= a[b]

なのだが，この書き方はなんだか簡略化しすぎていて理解が難しい気がする． おそらくNumpyだと，こういう演算が許されているということだと思うのだが．．．