如何在 Python 中为 CSV 文件添加页眉

CSV文件包含一种表格形式的数据，每一行都包含逗号分隔的数值。列可能包含属于不同数据结构的值。Python提供了大量的方法和模块来完成CSV文件与pandas数据框架的相互转换，反之亦然。CSV文件的页眉是一个分配给每一列的值的数组。它充当了数据的行头。最初，CSV文件被转换为一个数据框架，然后在数据框架中加入一个页眉。数据框架的内容再次被存储回CSV文件。

在这篇文章中，我们将在Python中为CSV文件添加一个标题。

方法#1：在to_csv()方法中使用header参数。

最初，以列表的形式创建一个头，然后使用to_csv()方法将该头添加到CSV文件中。

下面的CSV文件gfg.csv被用于操作。

# importing python package
import pandas as pd
  
# read contents of csv file
file = pd.read_csv("gfg.csv")
print("\nOriginal file:")
print(file)
  
# adding header
headerList = ['id', 'name', 'profession']
  
# converting data frame to csv
file.to_csv("gfg2.csv", header=headerList, index=False)
  
# display modified csv file
file2 = pd.read_csv("gfg2.csv")
print('\nModified file:')
print(file2)

输出:

CSV文件_gfg2.csv已经创建。

方法#2：使用DictWriter()方法

另一种使用DictWriter()的方法可以用来给CSV文件的内容添加一个头。fieldnames属性可以用来指定CSV文件的头，而delimiter参数用csv模块中给出的分隔符来分隔数值，以进行添加头的工作。然后在csvwriter对象上调用writeheader()方法，不传递任何参数。

# import required modules
import pandas as pd
import csv
  
# assign header columns
headerList = ['col1', 'col2', 'col3', 'col4']
  
# open CSV file and assign header
with open("gfg3.csv", 'w') as file:
    dw = csv.DictWriter(file, delimiter=',', 
                        fieldnames=headerList)
    dw.writeheader()
  
# display csv file
fileContent = pd.read_csv("gfg3.csv")
fileContent

输出:

新创建的CSV文件gfg3.csv是。

注意：此方法只适用于空的CSV文件。