PythonでリストからN個の最大要素を取得する方法

整数のリストが与えられたとき、その中からN個の大きな要素を取り出して新しいリストとして返すのが、ここでの課題です。本記事では、基本的なループ処理による方法から、Python標準ライブラリを活用した効率的な方法まで、サンプルコードとともに解説します。

例

入力 : [40, 5, 10, 20, 9]
N = 2
出力 : [40, 20]

アルゴリズム

1. 整数のリストと、取得する要素数Nを受け取ります。
2. N回のループを実行します。
3. 各ループでリスト内の最大値を探し、新しいリストに格納すると同時に元のリストから削除します。

実装コード

def Nnumberele(list1, N):
    new_list = []
    for i in range(0, N):
        max1 = list1[0]
        for j in range(len(list1)):
            if list1[j] > max1:
                max1 = list1[j]
        list1.remove(max1)
        new_list.append(max1)
    return new_list

# ドライバーコード
my_list = [12, 61, 41, 85, 40, 13, 77, 65, 100]
N = 4

# 関数の呼び出し
result = Nnumberele(my_list, N)
print("Largest numbers are", result)

出力

Largest numbers are [100, 85, 77, 65]

sorted()を使ったシンプルな方法

リストを降順にソートし、先頭のN個をスライスするだけでも同じ結果が得られます。コードが非常に簡潔になるため、実務ではよく使われるアプローチです。

my_list = [12, 61, 41, 85, 40, 13, 77, 65, 100]
N = 4
result = sorted(my_list, reverse=True)[:N]
print(result)  # [100, 85, 77, 65]

heapq.nlargest()を使った効率的な方法

標準ライブラリのheapqモジュールには、まさにこの目的のためのnlargest()関数が用意されています。リスト全体を完全にソートする必要がないため、Nが小さい場合には計算量がO(n log N)となり、大規模なデータでも高速に動作します。

import heapq

my_list = [12, 61, 41, 85, 40, 13, 77, 65, 100]
N = 4
result = heapq.nlargest(N, my_list)
print(result)  # [100, 85, 77, 65]

まとめ

手軽さを重視するならsorted()、パフォーマンスを重視するならheapq.nlargest()がおすすめです。一方、学習目的であれば、ループで最大値を繰り返し探索する基本手法も理解しておくと、アルゴリズムへの理解が深まります。