Pythonで最大差となる要素ペアを見つける方法

データ分析ではさまざまな課題に直面することがあります。本記事では、数値を要素として持つリストを対象に、その中から値の差が最も大きくなる要素のペアを見つける方法を解説します。

nlargest()を使った方法

このアプローチでは、まず itertools モジュールの combinations() 関数を使って、リスト内のすべての要素の組み合わせを生成します。続いて、各ペアについて2番目の要素を1番目の要素から引いた差の絶対値を計算し、heapq モジュールの nlargest() 関数を適用することで、差が最大となる上位のペアを取得します。この方法なら、必要な件数を指定して複数のペアをまとめて取得できるのが特徴です。

サンプルコード

from itertools import combinations
from heapq import nlargest

listA = [21, 14, 30, 11, 17, 18]

# 対象となるリスト
print("Given list : ", listA)

# nlargest() と combinations() を使用
res = nlargest(2, combinations(listA, 2),
               key=lambda sub: abs(sub[0] - sub[1]))

# 結果を出力
print("Pairs with maximum difference are : ", res)

実行結果

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

Given list : [21, 14, 30, 11, 17, 18]
Pairs with maximum difference are : [(30, 11), (14, 30)]

出力を見ると、最大の差を持つ (30, 11) と、次点の (14, 30) の2組が返されていることが分かります。

combinations() と max() を使った方法

こちらも基本的な考え方は上記と同じです。ただし、max() 関数を適用するため、結果として返されるのは差が最大となる1組のペアのみです。複数の候補ではなく、単一の答えだけが必要なケースに適した方法といえます。

サンプルコード

from itertools import combinations

listA = [21, 14, 30, 11, 17, 18]

# 対象となるリスト
print("Given list : ", listA)

# combinations() と lambda を使用
res = max(combinations(listA, 2), key=lambda sub: abs(sub[0] - sub[1]))

# 結果を出力
print("Pairs with maximum difference are : ", res)

実行結果

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

Given list : [21, 14, 30, 11, 17, 18]
Pairs with maximum difference are : (30, 11)

まとめ

combinations() で全ペアを生成し、キー関数で差を計算するという流れはどちらの方法も共通しています。上位N件のペアが欲しい場合は nlargest()、最大の1組だけでよい場合はシンプルに max() を使うとよいでしょう。状況に応じて使い分けることで、コードを簡潔かつ効率的に保てます。