¿Cómo agregar un encabezado a un archivo CSV en Python?

Un archivo CSV contiene un tipo de datos tabular donde cada fila contiene valores separados por comas. Las columnas pueden contener valores pertenecientes a diferentes estructuras de datos. Python proporciona una amplia gama de métodos y módulos para realizar la interconversión del archivo CSV a un marco de datos pandas y viceversa. Un encabezado del archivo CSV es una array de valores asignados a cada una de las columnas. Actúa como un encabezado de fila para los datos. Inicialmente, el archivo CSV se convierte en un marco de datos y luego se agrega un encabezado al marco de datos. El contenido del marco de datos se vuelve a almacenar en el archivo CSV. 

En este artículo, agregaremos un encabezado a un archivo CSV en Python.

Método n.º 1: usar el argumento del encabezado en el método to_csv() .

Inicialmente, cree un encabezado en forma de lista y luego agregue ese encabezado al archivo CSV usando el método to_csv() .

El siguiente archivo CSV  gfg.csv se utiliza para la operación:

# 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)

Producción: 

Se crea el archivo CSV gfg2.csv :

Método #2: Usando el método DictWriter()

Se puede usar otro enfoque para usar DictWriter() para agregar un encabezado al contenido de un archivo CSV. El atributo de nombres de campo se puede usar para especificar el encabezado del archivo CSV y el argumento del delimitador separa los valores por el delimitador dado en el módulo csv que se necesita para llevar a cabo la adición del encabezado. Luego se invoca el método writeheader () en el objeto csvwriter , sin pasar ningún argumento.

# 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

Producción:

El nuevo archivo CSV gfg3.csv creado es:

Nota: este método solo es aplicable a archivos CSV vacíos.