Как добавить заголовок к существующим CSV-файлам с помощью Python без переписать весь существующий контент - PullRequest
Новая
       32

Как добавить заголовок к существующим CSV-файлам с помощью Python без переписать весь существующий контент

0 голосов
/ 30 сентября 2018

Я пытаюсь добавить заголовок к моему существующему CSV-файлу, и в нем уже есть содержимое.Мне просто интересно, есть ли какой-нибудь фрагмент кода, который мог бы вставить строку заголовка вверху (например, ['name', 'age', 'salary', 'country'], не затрагивая содержимое.

Также этот код связан с API, поэтому я буду запускать его несколько раз. Так что просто интересно, можно ли определить, существует ли заголовок, чтобы избежать нескольких строк заголовка.

Спасибо и надеюсь, что всем вам хорошего дня!

1 Ответ

0 голосов
/ 30 сентября 2018

Ваш вопрос состоит из 2 частей:

1) Чтобы добавить заголовок в ваш CSV (если он не существует) Чтобы вставить строку заголовка, вы можете прочитать CSV с помощью следующей команды: 

df=pd.read_csv(filename, header=None, names=['name','age','salary','country'])

Чтобы создать csv со строкой заголовка без влияния на содержимое, вы можете использовать следующую команду: 

df.to_csv(new_file_with_header.csv, header=True)

2) Вторая часть немного хитрая.Чтобы определить, имеет ли ваш файл заголовок или нет, вам придется написать небольшой код.Я могу предоставить вам алгоритм.

явное чтение csv с заголовком 

df=pd.read_csv(filename.csv, header=None, names=['name','age','salary','country'])

Проверьте 1-ую строку 1-го столбца в вашем csv, если он содержит значение как «имя», то напишите csv из2-й ряд до конца, остальное напишите как 

temp_var=df['name'].iloc[0]

if (temp_var=='name'):
  df.iloc[1:].to_csv(new_file.csv)
else:
  df.to_csv(new_file.csv)

Надеюсь, это поможет !!

Спасибо, Рохан Ходаркар

...