Подтвердить что ты не робот

Panda Запись CSV - добавление против записи

Я хотел бы использовать pd.write_csv для записи "filename" (с заголовками), если "filename" не существует, иначе добавить к "filename" , если он существует. Если я просто использую команду:

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

Запись или добавление завершаются успешно, но кажется, что заголовок записывается каждый раз, когда происходит добавление.

Как я могу добавить заголовок, если файл не существует, и добавить без заголовка, если файл существует?

4b9b3361

Ответ 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)