web-dev-qa-db-ja.com

特定の文字列で始まるキーで辞書をスライスする

これは非常に簡単ですが、Pythonを使用してそれを実行する方法が大好きです。基本的に、辞書を指定すると、特定の文字列で始まるキーのみを含むサブ辞書が返されます。

» d = {'Apple': 1, 'Banana': 9, 'Carrot': 6, 'Baboon': 3, 'Duck': 8, 'Baby': 2}
» print slice(d, 'Ba')
{'Banana': 9, 'Baby': 2, 'Baboon': 3}

これは関数を使って行うのはかなり簡単です：

def slice(sourcedict, string):
    newdict = {}
    for key in sourcedict.keys():
        if key.startswith(string):
            newdict[key] = sourcedict[key]
    return newdict

しかし、より良い、賢い、より読みやすい解決策は確かにありますか？ここでジェネレータが役立ちますか？ （私はそれらを使用する十分な機会がありません）。

pythondictionaryironpythonslice
39
Aphex

これはどう：

in python 2.x：

def slicedict(d, s):
    return {k:v for k,v in d.iteritems() if k.startswith(s)}

In python 3.x：

def slicedict(d, s):
    return {k:v for k,v in d.items() if k.startswith(s)}
70
Mark Byers

機能的なスタイルで：

dict(filter(lambda item: item[0].startswith(string),sourcedict.iteritems()))

9
seriyPS

Python 代わりにitems()を使用します：

def slicedict(d, s):
    return {k:v for k,v in d.items() if k.startswith(s)}
3
gc5