Introducción al trabajo con archivos CSV

Los archivos CSV (valores separados por comas) son un formato habitual para almacenar datos tabulares. Python proporciona el módulo csv para leer, escribir y manipular archivos CSV de forma eficaz.

En esta lección, aprenderá a gestionar archivos CSV mediante el módulo integrado csv de Python.

¿Qué son los archivos CSV?

Los archivos CSV son archivos de texto que utilizan un delimitador (normalmente una coma) para separar los valores. Cada línea representa una fila y cada valor de la línea representa una columna.

Lectura de archivos CSV

Para leer un archivo CSV, utilice la función csv.reader. Esta función lee cada línea como una lista de valores:

# Reading a CSV file
import csv

with open("data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)  # Each row is a list of values

Escritura de archivos CSV

Para escribir en un archivo CSV, utilice la función csv.writer. Esta función permite escribir filas en un archivo CSV:

# Writing to a CSV file
import csv

with open("output.csv", "w", newline="") as file:
    writer = csv.writer(file)
    writer.writerow(["Name", "Age", "City"])
    writer.writerow(["Alice", 30, "New York"])
    writer.writerow(["Bob", 25, "London"])

Lectura de archivos CSV con encabezados

Utilice csv.DictReader para leer archivos CSV en diccionarios, cuyas claves son los encabezados de las columnas:

# Reading a CSV file with headers
import csv

with open("data.csv", "r") as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)  # Each row is a dictionary with column headers as keys

Escritura de archivos CSV con encabezados

Utilice csv.DictWriter para escribir diccionarios en archivos CSV con encabezados:

# Writing a CSV file with headers
import csv

with open("output.csv", "w", newline="") as file:
    fieldnames = ["Name", "Age", "City"]
    writer = csv.DictWriter(file, fieldnames=fieldnames)

    writer.writeheader()  # Write the header row
    writer.writerow({"Name": "Alice", "Age": 30, "City": "New York"})
    writer.writerow({"Name": "Bob", "Age": 25, "City": "London"})

Delimitadores personalizados en archivos CSV

Puede especificar un delimitador personalizado al leer o escribir archivos CSV. Por ejemplo, los archivos delimitados por tabulaciones utilizan una tabulación (\t) como delimitador:

# Using a custom delimiter
import csv

with open("data.tsv", "r") as file:
    reader = csv.reader(file, delimiter="\t")
    for row in reader:
        print(row)  # Each row is split by tab

Gestión de archivos CSV grandes

Para archivos CSV grandes, utilice un bucle que procese las filas de forma incremental y evite cargar todo el archivo en la memoria:

# Reading a large CSV file efficiently
import csv

with open("large_data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        # Process each row
        print(row)

Errores comunes con archivos CSV

Estos son algunos errores que debe evitar:

• Olvidar utilizar newline="" al escribir archivos CSV en Windows.
• Utilizar delimitadores incorrectos para archivos CSV no estándar.
• No gestionar los datos ausentes o con un formato incorrecto del archivo.

¿Qué aprendimos?

En esta lección, aprendió:

• Cómo leer y escribir archivos CSV mediante el módulo csv de Python.
• Cómo trabajar con archivos CSV que tienen encabezados mediante DictReader y DictWriter.
• Cómo gestionar archivos CSV grandes de forma eficaz.

¡Excelente trabajo! Pasemos al siguiente tema.