Pythonでネストされたリストから最大値を持つサブリストを検索する方法

Pythonでは、リストの要素として別のリスト(サブリスト)を持つことができます。本記事では、指定されたリストの中に含まれる複数のサブリストから、最大値を持つサブリストを見つける方法を2つのアプローチで解説します。

方法1: max関数とlambda式を組み合わせる

最もシンプルな方法は、max()関数とlambda式を組み合わせることです。lambda式でサブリスト内の比較対象となる要素(この例ではインデックス1の数値)を指定することで、その値が最大であるサブリストを取得できます。

コード例

listA = [['Mon', 90], ['Tue', 32], ['Wed', 120]]
# lambda式を使用して比較キーを指定
res = max(listA, key=lambda x: x[1])
# 結果を出力
print("Given List:\n", listA)
print("List with maximum value:\n ", res)

実行結果

上記のコードを実行すると、以下の出力が得られます。

Given List:
[['Mon', 90], ['Tue', 32], ['Wed', 120]]
List with maximum value:
['Wed', 120]

この例では、各サブリストの2番目の要素(90、32、120)を比較し、最大値120を持つ['Wed', 120]が返されています。

方法2: operatorモジュールのitemgetterを使う

もう一つの方法は、operatorモジュールのitemgetter()を使用する方法です。lambda式と同様に、itemgetter(1)でインデックス1の要素を抽出し、それをmax()関数のキーとして適用します。大量のデータを扱う場合、itemgetterはlambdaよりも高速に動作するという利点があります。

コード例

import operator
listA = [['Mon', 90], ['Tue', 32], ['Wed', 120]]
# itemgetterを使用して比較キーを指定
res = max(listA, key=operator.itemgetter(1))
# 結果を出力
print("Given List:\n", listA)
print("List with maximum value:\n ", res)

実行結果

上記のコードを実行すると、以下の出力が得られます。

Given List:
[['Mon', 90], ['Tue', 32], ['Wed', 120]]
List with maximum value:
['Wed', 120]

まとめ

どちらの方法でも同じ結果が得られます。簡潔さを重視するならlambda式、パフォーマンスを重視するならoperator.itemgetter()を選ぶのがおすすめです。状況に応じて使い分けることで、より効率的なPythonコードを書くことができます。