Нарезка словаря по ключам, начинающимся с определенной строки

Question

Это довольно просто, но я бы с удовольствием выбрал довольно питонский способ. По сути, при наличии словаря вернуть поддиректорий, содержащий только те ключи, которые начинаются с определенной строки.

» 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

Но наверняка есть более приятное, умное, более читаемое решение? Может ли здесь помочь генератор? (У меня никогда не было достаточно возможностей использовать их).

Answer 1

Как насчет этого:

в Python 2.x:

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

В Python 3.x:

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

Answer 2

В функциональном стиле:

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

Answer 3

В Python 3 используйте items() вместо:

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