Python如何在使用“with”打开文件后擦除文件

Question

我将脚本的循环迭代次数保存到检查点文件中：

with open('checkpoint.txt', 'w') as checkpoint_file:
    for i range(1000):
        //do stuff
        checkpoint_file.write(str(i) + '\n')

这将为每次迭代在我的文件中写入一个新行。

我希望在我中断脚本时只有一行包含最后一个迭代号，所以我想擦除“checkpoint.txt”文件的内容，然后在第一行写下我的迭代号（或者如果可能的话直接替换第一行）。

我知道如果我关闭文件然后用with open('checkpoint.txt', 'w') 再次打开它，它的内容将被删除，但为了提高效率，我想尽可能保持文件打开。

最好的方法是什么？

Answer 1

seeking 在每个 write 之前（并切换到行缓冲以避免需要单独使用 flush）会这样做：

# buffering=1 means you automatically flush after writing a line
with open('checkpoint.txt', 'w', buffering=1) as checkpoint_file:
    for i in range(1000):
        //do stuff
        checkpoint_file.seek(0)  # Seek back to beginning of file so next write replaces contents
        checkpoint_file.write(str(i) + '\n')

Answer 2

在每次写入之前寻找文件的开头。 见https://docs.python.org/2/library/stdtypes.html?highlight=seek#file.seek