python - Panda 的写入 CSV - 追加与写入

Question

如果“文件名”不存在，我想使用 pd.write_csv 写入“文件名”(带有标题)，否则如果存在，则附加到“文件名”。如果我只是使用命令:

     df.to_csv('filename.csv',mode = 'a',header ='column_names')

写入或追加成功，但似乎每次追加时都会写入标题。

如果文件不存在，我怎样才能只添加标题，如果文件存在，我怎样才能不添加标题？

Answer 1

不确定 pandas 中是否有方法，但检查文件是否存在将是一种简单的方法:

import os
# if file does not exist write header 
if not os.path.isfile('filename.csv'):
   df.to_csv('filename.csv', header='column_names')
else: # else it exists so append without writing the header
   df.to_csv('filename.csv', mode='a', header=False)