dictの上位**件を取り出す

Question

d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}

上記のdictが存在している場合にkey3まで取り出し、新たなdictにする方法がわかりません。

宜しくお願いいたします。

Answer 1

Python3.5以前ではdictの要素に値の順序の概念はありませんでした。

CPython3.6は処理系の仕様で、Python3.7は言語仕様でdictは順序を持つようになっています。

なので、最初にやるべきことは「自分がプログラムを動かすpythonのバージョンを確認すること」です。

もし辞書に順序があるバージョンを使っていれば、

d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}
result = {k:d[k] for k in list(d)[:3]}  # => {'key1': 1, 'key2': 14, 'key3': 47}

で希望の動作になると思います。

---

d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}

が文字通りkey1, key2,...というキーを持つ辞書なら

d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}
result = {"key"+str(i):d["key"+str(i)] for i in range(1,4)}  # => {'key1': 1, 'key2': 14, 'key3': 47}

という手もあります。

Comments

[hayataka2049]

上位**件が値の大小に基づく可能性もある・・・かな？　この回答は先頭からn件取るという内容ですが、意図と違っていたらコメントでご指摘ください

[limitman80]

バージョンは3.6になります。最初に書いておけばよかったです。もうしわけありません。
keyの部分は変化するのでそのままkey1とはならないです。

[hayataka2049]

であれば、念の為インタプリタを起動して
import sys;print(sys.implementation.name)
を見てください。cpythonと表示されれば回答の方法で問題ありません

Answer 2

（Python3.7でdictに順序を持つようになったのですね…）
「上位**件」が「降順に並べたときの最初からN個」といった意味ならば、以下はどうでしょうか。

# coding: utf-8
N = 3
d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}

# keyでソートしたときの上位N個
lists = sorted(d.items(), key=lambda x: x[0], reverse=True)
new_dict = {}
for i in range(N):
    new_dict[lists[i][0]] = lists[i][1]
print(new_dict)

# valueでソートしたときの上位N個
new_dict = {}
lists = sorted(d.items(), key=lambda x: x[1], reverse=True)
for i in range(N):
    new_dict[lists[i][0]] = lists[i][1]
print(new_dict)

Nが小さいならば、heapq.nlargest モジュールを使ったほうが高速です。

# coding: utf-8
import heapq

N = 3
d = {'key1': 1,'key2': 14,'key3': 47,'key4': 2,'key5': 90}

# keyでソートしたときの上位N個
lists = heapq.nlargest(N, d.items(), key=lambda x: x[0])
new_dict = {}
for l in lists:
    new_dict[l[0]] = l[1]
print(new_dict)

# valueでソートしたときの上位N個
lists = heapq.nlargest(N, d.items(), key=lambda x: x[1])
new_dict = {}
for l in lists:
    new_dict[l[0]] = l[1]
print(new_dict)