熊猫的CSV写入-附加与写入

发布于 2021-01-29 19:03:32

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

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

写入或追加成功,但似乎每次进行追加时都会写入标头。

如果文件不存在,如何只添加标题,如果文件存在,如何不添加标题而追加标题?

关注者
0
被浏览
45
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    不确定熊猫是否有办法,但是检查文件是否存在将是一种简单的方法:

    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)
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看