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 trabajar con archivos CSV mediante el módulo integrado csv de Python.

¿Qué son los archivos CSV?

Los archivos CSV son archivos de texto que usan 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, use 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, use la función csv.writer. Esta función le 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"])

Leer archivos CSV con encabezados

Use csv.DictReader para leer archivos CSV en diccionarios, donde las 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

Escribir archivos CSV con encabezados

Use 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 usan 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, use un bucle para procesar las filas progresivamente y evitar 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 los archivos CSV

Estos son algunos errores que debe evitar:

• Olvidar usar newline="" al escribir archivos CSV en Windows.
• Usar delimitadores incorrectos para archivos CSV no estándar.
• No gestionar los datos ausentes o con 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.