「NumPy」sortメソッドで二次元配列の列、行に対してソートするサンプル

2022年6月25日

環境
PyCharm 2021.3
Python 3.9.7

構文
二次元配列の変数名 = np.array([[要素1,要素2,xxx
二次元配列の場合はaxis=0で列に対してソート、axis=1で行に対してソートされます。
各列・各行の値が別々に並べ替えられます。

使用例

import numpy as np

cft = np.array([[22, 33, 100], [15, 200, 30], [345, 15, 26]])
print("二次元配列の要素")
print(cft)

print("axis=0で列に対してソートする結果")
sortcol = np.sort(cft, axis=0)
print(sortcol)

print("axis=1で行に対してソートされる結果")
sortrow = np.sort(cft, axis=1)
print(sortrow)

実行結果

二次元配列の要素
[[ 22  33 100]
 [ 15 200  30]
 [345  15  26]]
axis=0で列に対してソートする結果
[[ 15  15  26]
 [ 22  33  30]
 [345 200 100]]
axis=1で行に対してソートされる結果
[[ 22  33 100]
 [ 15  30 200]
 [ 15  26 345]]