Bus errorとSegmentation faultに困ったら見るブログ

物理の研究者による日々の研究生活のメモ書きです ( python/emacs/html/Japascript/シェルスクリプト/TeX/Mac/C言語/Linux/git/tmux/R/ポケモンGO)

【numpy, python3】あるarrayを別のarrayのソート順に従ってソートする(argsort)

実際使うことはなかったけど、今後必要な気もするのでメモ

2つのarrayがある、例えばxとyとする

xはデータのindex相当のもので、順番がぐちゃぐちゃになってる
yはデータの値
xをソートするときに、yの順番も同時に同じようにソートしてほしい

■ 参考 : argsort: ソートに使うインダイスを取得

■ 参考 : [Python]Numpyデータの並べ替え


argsortを使えばOK

index = np.argsort(x)
これで、xをソートしたときのindexを取得できる
x_new = x[index]
としたら、単純にxをソートしたことになる

indexを使って
y_new = y[index]
としたら、「xをソートするときに、yの順番も同時に同じようにソートしてほしい」が実現できる



ランキング参加中です

↓クリックしていただけると嬉しいです〜