Question

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

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

感谢您，并祝您生活愉快！

Answer 1

您的问题分为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'])

检查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)

希望这会有所帮助！

谢谢，罗汉·霍达卡（Rohan Hodarkar）