python - 如何使用python将标题添加到现有的csv文件而不重写所有现有内容

Question

我正在尝试向我现有的 csv 文件添加一个标题，并且其中已经有内容。我只是想知道是否有任何代码可以在顶部插入标题行（例如 ['name','age','salary','country'] 而不会影响内容。

此代码也连接到 API，因此我将多次运行它。所以只是想知道是否可以检测一个标题是否存在以避免多个标题行。

谢谢你，希望大家有个美好的一天！

score 0 · Accepted Answer

你的问题有两部分：

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'])

检查 csv 中的第一行第一列，如果它包含值作为“名称”，则将 csv 从第二行写入到末尾，否则按原样写入

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)

希望这可以帮助！！

谢谢，罗汉霍达卡

1 回答 1