Pythonで最大行要素を基準に行列を並べ替える方法

Pythonでは、行列（リストのリスト）を「各行の最大値」を基準にして並べ替えることができます。この記事では、max関数とsortメソッドを組み合わせて、行ごとの最大要素に従って行列を降順にソートする方法を解説します。

実装例

以下は、最大行要素を基準に行列を並べ替えるサンプルコードです。

def sort_max(row):
    return max(row)

my_list = [[15, 27, 18], [39, 20, 13], [13, 15, 56], [43, 13, 25]]

print("リストの内容：")
print(my_list)

my_list.sort(key=sort_max, reverse=True)

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

出力結果

リストの内容：
[[15, 27, 18], [39, 20, 13], [13, 15, 56], [43, 13, 25]]
並べ替え後の結果：
[[13, 15, 56], [43, 13, 25], [39, 20, 13], [15, 27, 18]]

コードの解説

• sort_maxという名前の関数を定義します。この関数は引数として1つの行を受け取り、その行の中で最も大きい要素をmax関数によって返します。

• 関数の外側で、複数のリストを含む入れ子のリスト（行列）を定義し、コンソールに表示します。

• sortメソッドを使用してリストを並べ替えます。このとき、key引数に先ほど定義したsort_max関数を指定することで、「各行の最大値」が比較基準となります。

• さらに、reverseパラメータをTrueに設定しているため、リストは降順（最大値が大きい行から順に）並べ替えられます。

• その結果、各行の最大値（18、39、56、43）に基づいて、[[13, 15, 56]] → [[43, 13, 25]] → [[39, 20, 13]] → [[15, 27, 18]] の順に出力されます。

lambdaを使った簡潔な書き方

別途関数を定義せず、lambda式を使えば同じ処理をより簡潔に記述できます。

my_list = [[15, 27, 18], [39, 20, 13], [13, 15, 56], [43, 13, 25]]

my_list.sort(key=lambda row: max(row), reverse=True)

print(my_list)
# [[13, 15, 56], [43, 13, 25], [39, 20, 13], [15, 27, 18]]

このように、key引数には任意の関数を渡せるため、行の合計値や平均値など、他の基準でも同様に並べ替えが可能です。状況に応じて使い分けるとよいでしょう。