使用 re.sub 时使用 '\g<index> 或 r'\index' 作为 dict 键？ [复制]

Question

例如，我有以下字符串：

s = 'Name: @name, ID: @id'

现在我想用re.sub() 替换@name 和@id。我知道我可以使用 group 来捕获一些字符串，然后使用 '\g<index>' 或 r'\index' 来使用它。

但现在我需要将它用作字典键，我有这个字典：

d = {'id': '20', 'name': 'Jon'}

我希望我能得到这个：

s = 'Name: Jon, ID: 20'

我也试过了：

>>> re.sub('@(\w+)', d[r'\1'], s)
Traceback (most recent call last):
  File "<input>", line 1, in <module>
KeyError: '\\1'

>>> re.sub('@(\w+)', d['\g<1>'], s)
Traceback (most recent call last):
  File "<input>", line 1, in <module>
KeyError: '\\g<1>'
>>> 

Answer 1

Python 提供了一个string.Template class（另见PEP 292），它可以格式化与您正在使用的字符串非常相似的字符串。默认情况下，string.Template 类将 $ 识别为占位符。如果您将其更改为@（通过子类化string.Template），那么您可以通过调用substitute 或safe_substitute 方法来执行替换：

import string
class MyTemplate(string.Template):
    delimiter = '@'

content = 'Name: @name, ID: @id'
d = {'id': '20', 'name': 'Jon'}
template = MyTemplate(content)
result = template.safe_substitute(**d)
print(result)

打印

Name: Jon, ID: 20

Answer 2

在这种情况下，您需要使用re.sub 的函数形式。对于您的基本用例，它可能很简单：

re.sub(r'@(\w+)', lambda m: d[m.group(1)], s)

如果逻辑更复杂，顶级def 是要走的路。基本上，根据re.sub docs，您传递callable 而不是str：

如果 repl 是一个函数，它会在每个不重叠的模式出现时调用。该函数采用单个匹配对象参数，并返回替换字符串。

Answer 3

如果可以选择以不同方式格式化您的字符串，您可以这样做：

>>> d = {'id': '20', 'name': 'Jon'}
>>> 'Name: {name}, ID: {id}'.format(**d)
'Name: Jon, ID: 20'