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

Question

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

此外，此代码已连接到 API，因此我将多次运行。所以只是想知道是否有可能检测是否存在 header 以避免多个 header 行。

谢谢大家，祝大家有个愉快的一天！

Answer 1

您的问题分为两部分：

1) 将 header 添加到您的 csv（当它不存在时）为了插入 header 行，您可以使用以下命令读取 csv：

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

要在不影响内容的情况下创建包含 header 行的 csv，您可以使用以下命令

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

2) 第二部分有点棘手。要推断您的文件是否包含 header，您必须编写一些代码。我可以给你算法。

使用 header

显式读取 csv

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

检查 csv 中的第 1 行第 1 列，如果它包含 'name' 的值，则将 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)

希望对您有所帮助！！

谢谢，罗汉霍达卡

how to add header to existing csv files using python without rewrite all the existing content