Pythonのある生活

Pythonの勉強を始めたインフラエンジニアです。インフラエンジニアって、プログラミング全く使わないΣ( ゚Д゚)ノ!ので、独学の勉強日記です。現在は、Progate、Aidemyを使っています。

ソートを行う

np.sort、argsort、sort

np.sort : 配列を数値の大きさ順に並び変える

argsort : 配列を数値の大きさ順に並び変え、元の配列で何番目に存在していたか出力する

sort : 引数に応じてソートを行う

 

【対話コード】

>>>import numpy as np

>>>arr = np.array([ [5, 8, 1], [4, 9, 3] ])

>>>print(np.sort(arr))

 [ [1, 5, 8],
   [3, 4, 9] ]

>>>print(arr.argsort())

 [ [2, 0, 1],
   [2, 0, 1] ]

>>>arr.sort(1)  #引数で1を指定すると行単位で要素をソートする

>>>print(arr)

[ [1 5 8]
 [3 4 9] ]