Python查找模式并替换整行

时间:2013-10-21 13:11:55

标签: python replace

def replace(file_path, pattern, subst):
   file_path = os.path.abspath(file_path)
   #Create temp file
   fh, abs_path = mkstemp()
   new_file = open(abs_path,'w')
   old_file = open(file_path)
   for line in old_file:
       new_file.write(line.replace(pattern, subst))
   #close temp file
   new_file.close()
   close(fh)
   old_file.close()
   #Remove original file
   remove(file_path)
   #Move new file
   move(abs_path, file_path)

我有这个函数来替换文件中的字符串。但我无法找到一种替代找到模式的整条线的好方法。

例如,如果我想替换一行包含:“John整天努力工作”使用模式“John”,替换将是“Mike没有那么努力”。

使用我当前的功能,我必须用模式写出整行以替换整行。

1 个答案:

答案 0 :(得分:7)

首先,您可以更改此部分:

for line in old_file:
    new_file.write(line.replace(pattern, subst))

进入这个:

for line in old_file:
    if pattern in line:
        new_file.write(subst)
    else:
        new_file.write(line)

或者你可以让它更漂亮:

for line in old_file:
    new_file.write(subst if pattern in line else line)