在python中同时读取和写入同一个文件

Question

是否可以同时读取和写入同一个文件？我有这段代码：

with open(total_f, 'w') as f:
    s= len(resu)
    #print s
    f.write ("the total number is {}\n".format(s))
    f.write('\n'.join(map(str, resu)))

with open(total_f ,'r') as f:
    content = f.read()
    count = content.count("FFFF")
    print count

但我需要他们在一起，因为我想把count的结果写在total_f中。

我试着像这样把读/写放在一起：

with open(total_f, 'rw') as f:# i want to read and write at the same time
    s= len(resu)
    content = f.read()
    count = content.count("FFFF")  
    #print s
    f.write ("the total number is {}\n".format(s))
    f.write ("the total number is {}\n".format(count))
    f.write('\n'.join(map(str, resu)))

但它仍然无法正常工作，所以我不确定这是否正确。

Answer 1

不清楚你想从书面文件中读取什么，所以我不保证这个答案会按原样工作：它旨在展示你可能使用的工作流程：

with open(total_f, 'w+') as f:
    s= len(resu)
    #print s
    f.write ("the total number is {}\n".format(s))
    pos_before_data = f.tell()
    f.write('\n'.join(map(str, resu)))

    f.seek(pos_before_data)
    content = f.read()
    count = content.count("FFFF")
    print count

关键是以'w+' 模式打开文件，如有必要，使用f.tell() 和f.seek() 进行导航。如果文件已经存在并且您不想覆盖它，则在'r+' 中打开。

Answer 2

你可以创建一个contextmanager:

import contextlib
@contextlib.contextmanager
def post_results(total_f, resu):
  with open(total_f ,'r') as f:
    content = f.read()
    count = content.count("FFFF")
  yield count
  with open(total_f, 'w') as f:
    s= len(resu)
    f.write ("the total number is {}\n".format(s))
    f.write('\n'.join(map(str, resu)))

with post_results('filename.txt', 'string_val') as f:
  print(f)

contextmanager 创建一个 enter 和 exit 序列，使函数能够自动执行所需的最终执行序列。在这种情况下，需要获取并打印count，然后将存储在count 中的值写入total_f。