python - 按以某个字符串开头的键对字典进行切片

Question

这很简单，但我喜欢一种漂亮的 Pythonic 方式。基本上，给定一个字典，返回仅包含以某个字符串开头的那些键的子字典。

» 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)}