【Python入門】リスト内のKより大きい値の個数を数える3つの方法

Pythonのリストの中から、特定の値(K)より大きい要素がいくつあるかを数える処理は、データ分析やアルゴリズムの問題など、多くの複雑な課題の基礎となる頻出の操作です。この記事では、代表的な3つの実装方法をサンプルコードとともに解説します。

方法1: forループを使う方法

最も基本的なアプローチは、forループでリストを順番に走査し、条件に合致する要素をカウントする方法です。処理の流れが直感的で分かりやすく、初心者にも理解しやすいのが特徴です。

# forループでKより大きい要素の個数を求める

# リストの初期化
test_list = [1, 7, 5, 6, 3, 8]

# 基準値kの設定
k = 4

# リストの表示
print("The list : " + str(test_list))

# forループでkより大きい数をカウント
count = 0
for i in test_list:
    if i > k:
        count = count + 1

# 結果の表示
print("The numbers greater than 4 : " + str(count))

方法2: リスト内包表記を使う方法

リスト内包表記を使えば、条件に合う要素だけを抽出した新しいリストを作成し、その長さをlen()で取得することで一括してカウントできます。コードが簡潔になり、Pythonらしい書き方として広く使われています。

# リスト内包表記でKより大きい要素の個数を求める

# リストの初期化
test_list = [1, 7, 5, 6, 3, 8]

# 基準値kの設定
k = 4

# リストの表示
print("The list : " + str(test_list))

# リスト内包表記でkより大きい数をカウント
count = len([i for i in test_list if i > k])

# 結果の表示
print("The numbers greater than 4 : " + str(count))

方法3: sum()関数を使う方法

Pythonではブール値(True/False)はそれぞれ1/0として扱われるため、sum()関数とジェネレータ式を組み合わせることで、条件を満たす要素の個数を1行で効率的に数えられます。メモリ効率も良く、大量のデータを扱う場合に有効です。

# sum()でKより大きい要素の個数を求める

# リストの初期化
test_list = [1, 7, 5, 6, 3, 8]

# 基準値kの設定
k = 4

# リストの表示
print("The list : " + str(test_list))

# sum()でkより大きい数をカウント
count = sum(i > k for i in test_list)

# 結果の表示
print("The numbers greater than 4 : " + str(count))

実行結果

いずれの方法でも、リスト[1, 7, 5, 6, 3, 8]の中で4より大きい要素(5, 6, 7, 8)は4つであるため、同じ結果が出力されます。

The list : [1, 7, 5, 6, 3, 8]
The numbers greater than 4 : 4
The list : [1, 7, 5, 6, 3, 8]
The numbers greater than 4 : 4
The list : [1, 7, 5, 6, 3, 8]
The numbers greater than 4 : 4

まとめ

リスト内のKより大きい値の個数を求めるには、forループ(可読性重視)、リスト内包表記(簡潔さ重視)、sum()関数(効率性重視)の3つの方法があります。用途やコードのスタイルに応じて使い分けるとよいでしょう。