如何使用 Python 将标头添加到现有 CSV 文件，而无需重写所有现有内容

Question

我正在尝试向现有的csv文件添加一个标题，并且其中已经有内容。我只是想知道是否有任何代码可以在不影响内容的情况下在顶部插入标题行(例如["姓名"，"年龄"，"薪水"，"国家"]。

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

谢谢你，祝大家有美好的一天！

Answer 1

您的问题分为两部分：

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中的第一行第一列，如果它包含值为"name"，则从第二行到结尾写入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)

希望这有帮助！！

谢谢 罗汉·霍达卡尔