Pythonで辞書から値の大きい上位3件を取得する方法

この記事では、Pythonの辞書（dict）から値が最も大きい上位3件の要素を取り出す方法について解説します。

課題

与えられた辞書から、値の大きい上位3件の「キー : 値」のペアを出力することを目標とします。

これを実現する主なアプローチとして、以下の2つの方法があります。

方法1：collections.Counter() を使う

標準ライブラリの collections モデルにある Counter クラスを利用すると、most_common() メソッドで簡単に上位N件を取得できます。

コード例

# collectionsモジュールをインポート
from collections import Counter

# 辞書
my_dict = {'T': 23, 'U': 22, 'O': 20, 'R': 32, 'S': 99}

k = Counter(my_dict)

# 値の大きい上位3件を取得
high = k.most_common(3)

print("値の大きい上位3件の辞書:")
print("キー : 値")
for i in high:
    print(i[0], " : ", i[1])

実行結果

値の大きい上位3件の辞書:
キー : 値
S  : 99
R  : 32
T  : 23

most_common(n) メソッドは、出現回数の多い順にn個の要素とそのカウントをリスト形式で返します。辞書を Counter に渡すと各キーの値がそのままカウントとして扱われるため、結果的に値の降順でソートされたデータが得られます。

方法2：heapq.nlargest() を使う

heapq モジュールの nlargest() 関数を使うと、ソート処理を明示的に書かずに上位N件を効率的に取得できます。

コード例

# heapqモジュールからnlargestをインポート
from heapq import nlargest

# 辞書
my_dict = {'T': 23, 'U': 22, 'O': 20, 'R': 32, 'S': 99}

ThreeHighest = nlargest(3, my_dict, key=my_dict.get)

print("値の大きい上位3件の辞書:")
print("キー : 値")
for val in ThreeHighest:
    print(val, " : ", my_dict.get(val))

実行結果

値の大きい上位3件の辞書:
キー : 値
S  : 99
R  : 32
T  : 23

nlargest() 関数は3つの引数を受け取ります。第1引数は取得する要素数、第2引数は対象となる辞書、第3引数の key には並び替えの基準となる関数を指定します。ここでは my_dict.get を渡すことで、各キーに対応する値を基準に降順ソートを行っています。

まとめ

この記事では、collections.Counter の most_common() メソッドと、heapq.nlargest() 関数という2つのアプローチを使って、辞書から値の大きい上位3件を取得する方法を学びました。どちらも標準ライブラリだけで実現できる手軽な手法なので、データの規模や用途に応じて使い分けるとよいでしょう。