按照以特定字符串开头的键对字典进行切片

6 浏览
0 Comments

按照以特定字符串开头的键对字典进行切片

这很简单，但我希望有一种漂亮的、具有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

但肯定有一种更好、更聪明、更易读的解决方案吧？在这里，生成器能起到帮助作用吗？(我从来没有足够的机会使用它们)。

0