CSV. Datos estructurados (III)
Acceso y descarga de csv
Vamos a centrarnos en este script en el trabajo con un archivo .csv, pensando no en el acceso (tema ya tratado en esta entrada), sino en tener disponible su contendio desde el script. Eso sí, seguiremos usando el módulo propio CSV, si bien ahora en la opción de diccionario.
Para que se entienda mejor el proceso a seguir, empezaremos por el acceso al archivo .csv. La diferencia principal respecto al script parejo de la entrada anterior es que sustituímos lector = csv.reader(archivo, delimiter=',') por lector_dict = csv.DictReader(archivo). El resto derivan de ésta.
import csv
archivo_nombre = 'datos/datos.csv' # 1. Definimos la ruta del archivo
try:
with open(archivo_nombre, mode='r', encoding='utf-8') as archivo: # 2. Abrimos el archivo en modo lectura ('r')
lector_dict = csv.DictReader(archivo) # 3. Creamos el lector de diccionarios
print(f"Leyendo datos de: {archivo_nombre}\n" + "-"*30)
for fila in lector_dict: # 4. Recorremos cada fila (cada fila es un diccionario)
nombre = fila['Nombre']
edad = fila['Edad']
print(f"Usuario: {nombre} | Edad: {edad}")
except FileNotFoundError: #Control de errores
print(f"Error: El archivo '{archivo_nombre}' no existe.")
except KeyError as e:
print(f"Error: No se encontró la columna {e} en el archivo.")
A pesar de las diferencias, ambos archivo comparten la misma limitación: la base de datos sólo está disponible dentro del ámbito de with open(), pero fuera no podemos acceder a los datos, así que si nos interesa deberemos trasladarlos a una colección de datos del propio script, concretamente a una lista de diccionarios.
import csv
archivo_nombre = 'datos/datos.csv' # 1. Definimos la ruta del archivo
datos_internos = [] # Lista para almacenar la base de datos en memoria
try:
with open(archivo_nombre, mode='r', encoding='utf-8') as archivo: # 2. Abrimos el archivo en modo lectura ('r')
lector_dict = csv.DictReader(archivo) # 3. Creamos el lector de diccionarios
print(f"Leyendo datos de: {archivo_nombre}\n" + "-"*30)
for fila in lector_dict: # 4. Recorremos cada fila (que es un diccionario)
nombre = fila['Nombre']
edad = int(fila['Edad'])
datos_internos.append(fila)
print(f"Usuario: {nombre} | Edad: {edad}")
print(f"--- Carga finalizada ---")
print(f"Se han almacenado {len(datos_internos)} registros en el diccionario interno.")
if datos_internos: # Ejemplo: Acceder al primer registro almacenado
primero = datos_internos[0]
print(f"Primer registro en memoria: {primero['Nombre']} tiene {primero['Edad']} años.")
except FileNotFoundError: #Control de errores
print(f"Error: El archivo '{archivo_nombre}' no existe.")
except KeyError as e:
print(f"Error: No se encontró la columna {e} en el archivo.")
#Ejemplo de acceso a la estructura del diccionario ---------------------------------------------------------------------------------------
registro_ejemplo = datos_internos[0] # Tomamos el primer diccionario de nuestra lista para el ejemplo
print(f"Claves: {list(registro_ejemplo.keys())}") # A. Identificar solo las CLAVES
print(f"Valores: {list(registro_ejemplo.values())}") # B. Identificar solo los VALORES
Lo que nos permite importar al script el contenido del archivo .csv empieza aquí (datos_internos = []) y se concreta en estas dos líneas:
for fila in lector_dict:----> Recorre el archivo .csv----datos_internos.append(fila)----> Añade contenido a la lista (de diccionarios)datos_internos
Ellas hacen posible el correcto acceso a los datos mediante instrucciones como print(f"Claves: {list(registro_ejemplo.keys())}"), que muestra las claves del diccionario.
Cierto que la consecuencia es un incremento de la carga de memoria, pero la ventaja en rapidez de funcionamiento compensa cuando necesitamos realizar diferentes acciones sobre los datos desde un mismo script. Tal es el caso del que sigue y con el que finalizamos esta entrada: un script mediante el que accedemos a una base de datos, la cargamos en memoria y realizamso sobre ella un conjunto de acciones que aquí se concretan en seleccionar y filtrar registros.
import csv
# --- FASE 1: CARGA DE DATOS ---------------------------------------------------
ruta_archivo = 'datos/datos.csv'
datos_internos = []
try:
with open(ruta_archivo, mode='r', encoding='utf-8') as archivo:
lector = csv.DictReader(archivo)
for fila in lector:
fila['Edad'] = int(fila['Edad']) # Convertimos el valor de [Edad] a entero para poder filtrar
datos_internos.append(fila)
print(f"Sistema listo. {len(datos_internos)} registros cargados.\n")
except FileNotFoundError:
print("Error: No se encontró el archivo en D:/. Por favor, créalo primero.")
exit() # Finaliza el script si no hay datos
# --- FASE 2: OPCIONES ------------------------------------------------------------
while True:
print("\n===============================")
print(" GESTOR DE CONSULTAS")
print("===============================")
print("1. Buscar por NOMBRE (Coincidencia parcial)")
print("2. Filtrar por EDAD (Exacta o Rango)")
print("3. Salir")
opcion = input("\nSeleccione una opción: ").strip()
if opcion == '3':
print("Saliendo del sistema...")
break
# --- OPCIÓN 1: BÚSQUEDA POR NOMBRE ---
if opcion == '1':
termino = input("Escriba el nombre o parte de él: ").strip().lower()
encontrados = [r for r in datos_internos if termino in r['Nombre'].lower()]
print(f"\nResultados para '{termino}':")
if encontrados:
for e in encontrados:
print(f"ID: {e['Nombre']} | Edad: {e['Edad']}")
else:
print("No se encontraron coincidencias.")
# --- OPCIÓN 2: FILTRO POR EDAD ---
elif opcion == '2':
print("\n--- Filtro de Edad ---")
print("A. Edad exacta")
print("B. Rango (Mínimo y Máximo)")
sub = input("Elija modalidad (A/B): ").strip().upper()
resultados = []
try:
if sub == 'A':
objetivo = int(input("Edad a buscar: "))
resultados = [r for r in datos_internos if r['Edad'] == objetivo] # Identificamos si el valor de la clave 'Edad' coincide
elif sub == 'B':
v_min = int(input("Edad mínima: "))
v_max = int(input("Edad máxima: "))
resultados = [r for r in datos_internos if v_min <= r['Edad'] <= v_max] # Filtro por intervalo de valores
if resultados:
print(f"\nSe hallaron {len(resultados)} personas:")
for r in resultados:
print(f"• {r['Nombre']} - {r['Edad']} años")
else:
print("No hay registros en ese rango.")
except ValueError:
print("Error: Por favor, ingrese solo números enteros para la edad.")
else:
print("Opción no válida. Por favor, marque 1, 2 o 3.")
print("\nPrograma finalizado correctamente.")
NOTA: El archivo .csv de prueba es necesario para el correcto funcionamiento de estos tres script, aunque siempre los puedes adaptar para que funcionen con un .csv diferente. No obstante, si prefieren no hacerlo, aquí lo puedes descargar.
