热门标签:
Q:

Python,使用通配符访问字典

我有一个包含项目的字典,其中键是某种正则表达式。 我正在寻找一个函数,每当传递匹配字符串时返回匹配项列表

d = {'a.b': item1, 'b.c':item2}

func(d,'a1b')
>>> [item1]

func(d,'b2c')
>>> [item2]

有没有pythonic的方式来做到这一点? 我能想出的唯一解决方案是:

import re
def func(d, event):
    res = list()
    for key, item in d.iteritems():
        if re.match(key, event):
            res.append(item)
    return res
原网址
A:

您可以创建一个包装dictionary类的类来为您执行此操作:

import re

class RegexDict(dict):

    def get_matching(self, event):
        return (self[key] for key in self if re.match(key, event))

使用示例:

>>> d = {'a.b': 'item1', 'b.c': 'item2'}
>>> rd = RegexDict(d)
>>> for o in rd.get_matching('a1b'):
    print o


item1

这节省了您必须依赖外部函数,它保持开销小,并且更加健壮。

您还可以添加另一个函数,该函数将列表(或iter)作为输入并返回所有匹配的值:

    def get_all_matching(self, events):
        return (match for event in events for match in self.get_matching(event))

>>> for o in rd.get_all_matching(['a1b', 'b2c']):
    print o


item1
item2

所有回答

共 1 条

author avatar

您不需要循环这些值。 只要钥匙就足够了:

import re
def func(d, event):
    res = []
    for key in d.keys():
        if re.match(key, event):
            res.append(d[key])
    return res

这是一个生成器版本。 这个更快,因为它不存储整个列表,而只在需要时生成它:

def func(d, event):
    for key in d.keys():
        if re.match(key, event):
            yield d[key]

还有一个班轮:

#1 List Comprehension
return [d[key] for key in d.keys() if re.match(key, event)]
#2 Generator Expression
return (d[key] for key in d.keys() if re.match(key, event))

相似问题