Lectura y escritura de archivos CSV en Python

Los archivos CSV (valores separados por comas) se utilizan ampliamente para el almacenamiento e intercambio de datos. En Python, puedes leer y escribir fácilmente en archivos CSV mediante el módulo csv integrado. Este artículo te guiará a través de los conceptos básicos para trabajar con archivos CSV en Python.

Lectura de archivos CSV

Para leer un archivo CSV, puede utilizar la clase csv.reader del módulo csv. A continuación, se muestra un ejemplo sencillo:

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

En este ejemplo, example.csv se abre en modo de lectura y csv.reader se utiliza para analizar el archivo. Cada fila del archivo CSV se imprime como una lista de valores.

Escritura de archivos CSV

Para escribir datos en un archivo CSV, utilice la clase csv.writer. A continuación, se muestra cómo escribir una lista de filas en un archivo CSV:

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

En este ejemplo, se crea o sobrescribe output.csv con los datos especificados. Se utiliza el método writer.writerows() para escribir varias filas a la vez.

Conclusión

Trabajar con archivos CSV en Python es sencillo si se utiliza el módulo csv. Puede leer y escribir fácilmente en archivos CSV, lo que facilita el manejo de datos en este formato común.