python - 打开文件，读取它，处理，然后写回 - Python 中最短的方法

Question

我想对一个文件做一些基本的过滤。读它，做处理，写回来。

我不是在寻找“打高尔夫球”，而是想要最简单、最优雅的方法来实现这一目标。我想到了:

from __future__ import with_statement

filename = "..." # or sys.argv...

with open(filename) as f:
    new_txt = # ...some translation of f.read() 

open(filename, 'w').write(new_txt)

with 语句使事情变得更短，因为我不必显式打开和关闭文件。

还有其他想法吗？

Answer 1

实际上，使用 fileinput 更简单的方法是使用 inplace 参数:

import fileinput
for line in fileinput.input (filenameToProcess, inplace=1):
    process (line)

如果您使用 inplace 参数，它会将 stdout 重定向到您的文件，这样如果您进行打印，它就会写回您的文件。

此示例将行号添加到您的文件中:

import fileinput

for line in fileinput.input ("b.txt",inplace=1):
    print "%d: %s" % (fileinput.lineno(),line),