python - 对文件中的行使用正则表达式

Question

我有一个文本文件，应该像这样更改:

Go through the lines and If you see print var[number or string] change it to print ( var[number or string])

对于输入:

a = [1,2,3]

print a[1]

输出必须是:

a = [1,2,3]

print(a[1])

我尝试过这个:

import re
with open('file.txt', 'r') as f:
     data = f.read()

newline = re.sub(r"^print\s(.+)", r"print(\1)", data)

with open('file.txt', 'w') as f:
f.write(newline)

仅当 print 位于第一行时才有效。我应该如何检查所有行并更改它们？

Answer 1

您需要使用 M 标志将行的开头与 ^ 匹配，而不是与文本的开头匹配。

re.sub("^a", "c", "abba\nabba", flags=re.M)
'cbba\ncbba'