Panda Запись CSV - добавление против записи
Я хотел бы использовать pd.write_csv для записи "filename" (с заголовками), если "filename" не существует, иначе добавить к "filename" , если он существует. Если я просто использую команду:
df.to_csv('filename.csv',mode = 'a',header ='column_names')
Запись или добавление завершаются успешно, но кажется, что заголовок записывается каждый раз, когда происходит добавление.
Как я могу добавить заголовок, если файл не существует, и добавить без заголовка, если файл существует?
Ответы
Ответ 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)