通过以特定string开头的键来切分字典

这很简单,但我喜欢这样做的一个漂亮,pythonic的方式。 基本上,给定一个字典,返回只包含那些以某个string开头的键的子字典。

» 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 

但肯定有更好,更聪明,更可读的解决scheme? 发电机可以帮助吗? (我从来没有足够的机会使用这些)。

这个怎么样:

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

function风格:

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

在Python 3中,使用items()来代替:

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