Crear carpetas de archivos y guardar gráficos en ellas con Python

El siguiente código es un ejemplo de cómo puedes crear carpetas (si no existe ya) y guardar un gráfico que hayas creado dentro de dicha carpeta.

import matplotlib.pyplot as plt
import os

def crear_grafico(datos_x, datos_y, titulo, nombre_archivo, carpeta):
  """
  Función que genera un gráfico de líneas y lo guarda en una carpeta.

  Args:
    datos_x: Lista de valores para el eje X.
    datos_y: Lista de valores para el eje Y.
    titulo: Título del gráfico.
    nombre_archivo: Nombre del archivo donde se guardará el gráfico.
	carpeta: ruta donde quedará guardado el gráfico
  Returns:
    None
  """

  # Crea la figura y el eje
  plt.figure()
  ax = plt.axes()

  # Crea el gráfico de líneas
  ax.plot(datos_x, datos_y)

  # Configura el título y las etiquetas de los ejes
  ax.set_title(titulo)
  ax.set_xlabel("Eje X")
  ax.set_ylabel("Eje Y")

  # Crea la carpeta si no existe
  if not os.path.exists(carpeta):
    os.makedirs(carpeta)

  # Guarda el gráfico en la carpeta
  plt.savefig(carpeta+"/"+nombre_archivo)

# Ejemplo de uso
datos_x = [1, 2, 3, 4, 5]
datos_y = [2, 4, 5, 4, 2]
titulo = "Ejemplo de gráfico de líneas"
nombre_archivo = "grafico.png"
carpeta = "graficos"

crear_grafico(datos_x, datos_y, titulo, nombre_archivo, carpeta)

Explicación del código:

1. Importa las bibliotecas necesarias:
   • matplotlib.pyplot: Para generar el gráfico.
   • os: Para crear la carpeta si no existe.
2. Define la función crear_grafico:
   • Recibe como argumentos:
     • datos_x: Lista de valores para el eje X.
     • datos_y: Lista de valores para el eje Y.
     • titulo: Título del gráfico.
     • nombre_archivo: Nombre del archivo donde se guardará el gráfico.
   • Crea la figura y el eje.
   • Genera el gráfico de líneas.
   • Configura el título y las etiquetas de los ejes.
   • Crea la carpeta si no existe.
   • Guarda el gráfico en la carpeta.
3. Ejemplo de uso:
   • Se crean listas de datos para el eje X y el eje Y.
   • Se define el título del gráfico.
   • Se define el nombre del archivo donde se guardará el gráfico.
   • Se llama a la función crear_grafico para generar y guardar el gráfico.

Personalización del gráfico:

• Puedes modificar el tipo de gráfico utilizando la función plt.plot. Por ejemplo, para crear un gráfico de barras, puedes usar plt.bar(datos_x, datos_y).
• Puedes agregar más opciones de personalización al gráfico, como colores, marcadores, leyenda, etc.
• Puedes modificar el nombre del archivo y la ubicación de la carpeta donde se guardará el gráfico.

Nota:

• Este código es un ejemplo básico. Puedes modificarlo para adaptarlo a tus necesidades específicas.
• Asegúrate de tener instalado matplotlib antes de ejecutar el código. Puedes instalarlo usando el comando pip install matplotlib.