这很简单,但我喜欢这种漂亮,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
但肯定有一个更好,更聪明,更易读的解决方案?发电机可以帮忙吗? (我从来没有足够的机会使用它们)。
答案 0 :(得分:70)
这个怎么样:
在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)}
答案 1 :(得分:9)
功能风格:
dict(filter(lambda item: item[0].startswith(string),sourcedict.iteritems()))
答案 2 :(得分:3)
在Python 3中使用items()
代替:
def slicedict(d, s):
return {k:v for k,v in d.items() if k.startswith(s)}