Pythonで行列を行ごとの中央値を基準に並べ替える方法

リストのリスト（行列）を、各行の中央値を基準にして並べ替えたい場合があります。Pythonでは、標準ライブラリ statistics モジュールが提供する median 関数を利用したキー関数を定義することで、この処理を簡単に実現できます。

なお、中央値とはデータを昇順に並べたときに中央に位置する値のことで、外れ値の影響を受けにくい指標として知られています。

サンプルコード

from statistics import median

def median_row(row):
    return median(row)

my_list = [[43, 14, 27], [13, 27, 24], [32, 56, 18], [34, 62, 55]]

print("元のリスト：")
print(my_list)

my_list.sort(key=median_row)

print("並べ替え後の結果：")
print(my_list)

実行結果

元のリスト：
[[43, 14, 27], [13, 27, 24], [32, 56, 18], [34, 62, 55]]
並べ替え後の結果：
[[13, 27, 24], [43, 14, 27], [32, 56, 18], [34, 62, 55]]

コードの解説

• まず、中央値を計算するために statistics モジュールから median 関数をインポートします。

• median_row という名前の関数を定義します。この関数は1つの行（リスト）を引数として受け取り、median 関数を使ってその行の中央値を戻り値として返します。

• 入れ子になったリスト（行列）を定義し、コンソールに表示して元の状態を確認します。

• sort メソッドを呼び出し、key 引数に先ほど定義した median_row 関数を指定します。これにより、各行の中央値を比較基準としてリスト全体が昇順に並べ替えられます。

• 最後に、並べ替え後の結果をコンソールに出力します。

補足：sorted() を使う方法

元のリストを変更せずに、新しい並べ替え済みリストを作成したい場合は、組み込み関数 sorted() を使うこともできます。

sorted_list = sorted(my_list, key=median_row)

list.sort() はリスト自体を破壊的に変更するのに対し、sorted() は元のリストを保持したまま新しいリストを返すため、用途に応じて使い分けるとよいでしょう。