如果正则表达式匹配则替换字符串，并用匹配其他正则表达式的其他东西替换

Question

在满足某些条件后，我尝试创建dict。 这是代码片段：

def dic_gen(exc):
    param_errors = {re.sub(r"sss_", r"aaa_",err.name): err.address for err in exc.errors }
    param_errors["status"] = "ERROR"
    return param_errors

上面的代码正在做的是检查err.name是否具有sss_，然后将其删除并创建字典。 现在，我还想添加另一个条件，如果它具有“ ttt_”，然后将其替换为“ bbb_”，那么使用re.sub可以吗？ 或最有效的方法是什么？

谢谢，

Answer 1

我不知道您打算如何处理re，但我会使用：

abc = "Something_sss whatever"
result = abc.replace("_sss","_ttt")
print result

结果：

"Something_ttt whatever"

Answer 2

通过传递re.sub()一个函数而不是替换字符串，您可以像这样进行操作：

def func(matchobj):
    return 'aaa_' if matchobj.group(0) == 'sss_' else 'bbb_'

def dic_gen(exc):
    param_errors = {re.sub(r'(sss_)|(ttt_)', func, err.name):
                       err.address for err in exc.errors}
    param_errors["status"] = "ERROR"
    return param_errors

由于该函数只是一个表达式，您可以使用它做一个lambda并避免使用外部函数定义（尽管它使代码的可读性降低）：

def dic_gen(exc):
    param_errors = {re.sub(r'(sss_)|(ttt_)', 
        lambda mo: 'aaa_' if mo.group(0) == 'sss_' else 'bbb_', err.name):
        err.address for err in exc.errors}
    param_errors["status"] = "ERROR"
    return param_errors