Pythonでネストされたリストの各サブリストから最大値を取得する方法

Pythonでは、リストの中にリストが入った「ネストされたリスト（リストのリスト）」を扱うことがよくあります。このようなデータ構造において、内側の各サブリストから最大値を取り出したいケースは多々あります。

本記事では、代表的な2つの方法——リスト内包表記とmax()関数を組み合わせる方法、そしてmap()関数とmax()関数を組み合わせる方法——を紹介します。

方法1：max関数とinを使ったリスト内包表記

最もシンプルで可読性が高いのが、リスト内包表記を使う方法です。for文でサブリストを1つずつ取り出しながら、それぞれにmax()関数を適用することで、各サブリストの最大値をまとめて取得できます。

コード例

Alist = [[10, 13, 454, 66, 44], [10, 8, 7, 23]]
# 元となるリスト
print("元のリスト:\n ", Alist)
# 各サブリストにmaxを適用
res = [max(elem) for elem in Alist]
# 最大値を出力
print("各サブリストの最大値:\n ", res)

実行結果

上記のコードを実行すると、次のような出力が得られます。

元のリスト:
[[10, 13, 454, 66, 44], [10, 8, 7, 23]]
各サブリストの最大値:
[454, 23]

最初のサブリスト [10, 13, 454, 66, 44] の最大値は 454、2番目のサブリスト [10, 8, 7, 23] の最大値は 23 であるため、結果として [454, 23] という新しいリストが生成されます。

方法2：map関数とmax関数を組み合わせる

もうひとつの方法は、map()関数を使うアプローチです。map(max, Alist)と書くことで、Alist内の各サブリスト（イテラブルな要素）に対して順番にmax()関数が適用されます。戻り値はmapオブジェクトなので、list()で明示的にリストへ変換する点に注意してください。

コード例

Alist = [[10, 13, 454, 66, 44], [10, 8, 7, 23]]
# 元となるリスト
print("元のリスト:\n ", Alist)
# mapで各サブリストにmaxを適用
res = list(map(max, Alist))
# 最大値を出力
print("各サブリストの最大値:\n ", res)

実行結果

実行すると、先ほどと同じ結果が得られます。

元のリスト:
[[10, 13, 454, 66, 44], [10, 8, 7, 23]]
各サブリストの最大値:
[454, 23]

どちらの方法を選ぶべきか？

• リスト内包表記：Pythonicで読みやすく、条件を追加したい場合（例：空リストのスキップなど）にも柔軟に対応できるため、一般的にはこちらが推奨されます。
• map関数：既存のコードとの一貫性や、関数型スタイルを好む場合に有効です。単純な処理ではパフォーマンスもほぼ同等です。

いずれの方法でも、ネストされたリストから各サブリストの最大値を簡潔に抽出できるため、用途やチームのコーディング規約に応じて使い分けるとよいでしょう。