python - 使用正则表达式替换特定匹配项

Question

我想使用正则表达式执行替换，不是针对所有匹配项，而是仅针对特定匹配项。但是，re.sub 替代了所有匹配项。我该怎么做？

这是一个例子。比如说，我有一个包含以下内容的字符串:

FOO=foo1
BAR=bar1
FOO=foo2
BAR=bar2
BAR=bar3

我想做的是:

re.sub(r'^BAR', '#BAR', s, index=[1,2], flags=re.MULTILINE)

得到下面的结果。

FOO=foo1
BAR=bar1
FOO=foo2
#BAR=bar2
#BAR=bar3

Answer 1

您可以将替换函数传递给 re.sub 以跟踪计数并检查是否应替换给定的索引:

import re

s = '''FOO=foo1
BAR=bar1
FOO=foo2
BAR=bar2
BAR=bar3'''

i = 0
index = {1, 2}

def repl(x):
    global i
    if i in index:
        res = '#' + x.group(0)
    else:
        res = x.group(0)

    i += 1
    return res

print re.sub(r'^BAR', repl, s, flags=re.MULTILINE)

输出:

FOO=foo1
BAR=bar1
FOO=foo2
#BAR=bar2
#BAR=bar3