Python:以模式开头替换文件中的行

时间:2014-09-09 16:02:41

标签: python

在文本文件中有这一行

initial_mass = unknown_int 

我想用

替换该行
initial_mass = known_int 

顾名思义,我不知道初始值,但我想用已知值替换它。使用Python,我如何搜索以' initial_mass'开头的行,然后用" initial_mass = known_int'?

替换整行。

我正在思考

import fileinput
    for line in fileinput.FileInput(filename, inplace=True):
    line=line.replace(old, new)
    print(line),  

如何将old设置为包含initial_mass = uknown_int的行? 我看过startswith(),但我不知道如何使用它来获得我想要的东西。我也试过正则表达式,发现它完全令人困惑。

文件不是特别大,但我需要多次迭代这个过程。

1 个答案:

答案 0 :(得分:6)

您不需要知道old是什么;重新定义整行:

import sys
import fileinput
for line in fileinput.input([filename], inplace=True):
    if line.strip().startswith('initial_mass = '):
        line = 'initial_mass = 123\n'
    sys.stdout.write(line)