import matplotlib.pyplot as plt

# Datos de los meses y ahorro en miles de pesos
meses = ['Enero', 'Febrero', 'Marzo', 'Abril', 'Mayo', 'Junio', 
         'Julio', 'Agosto', 'Septiembre', 'Octubre', 'Noviembre', 'Diciembre']

ahorro_cuernavaca = [
    1.45, 1.48, 1.99, 1.73, 1.86, 
    1.86, 2.09, 1.42, 1.42, 1.50, 
    1.72, 1.43
]

ahorro_mexicali = [
    1.14, 1.28, 1.82, 2.28, 2.70, 
    3.87, 3.47, 2.47, 2.37, 1.62, 
    1.26, 0.84
]

# Crear una figura con un solo eje Y y ajustar el tamaño de la figura
fig, ax = plt.subplots(figsize=(12, 8))

# Graficar ahorro en Cuernavaca
ax.plot(meses, ahorro_cuernavaca, marker='o', color='blue', linestyle='-', linewidth=2, label='Localidad de Cuernavaca (MWh/USD)')
# Graficar ahorro en Mexicali
ax.plot(meses, ahorro_mexicali, marker='s', color='red', linestyle='-', linewidth=2, label='Localidad de Mexicali (MWh/USD)')

# Configuración del eje Y
ax.set_ylabel('Dinero ahorrado (MWh/USD)', fontsize=14)
ax.tick_params(axis='both', labelsize=12)

# Configuración del eje X
ax.set_xlabel('Meses', fontsize=14)
ax.set_xticks(meses)

# Ajuste de límites del eje Y
ax.set_ylim(0, max(max(ahorro_cuernavaca), max(ahorro_mexicali)) + 1)  # Aumenta el límite superior para ahorro

# Título general
plt.title('Dinero ahorrado por mes para las localidades analizadas', fontsize=20)

# Leyenda ubicada dentro del gráfico en la parte superior izquierda
ax.legend(loc='upper left', fontsize=12)

# Eliminar el grid
ax.grid(False)

# Acentuar el marco del gráfico en todos los lados
for spine in ax.spines.values():
    spine.set_edgecolor('black')
    spine.set_linewidth(2)  # Define el grosor del marco

# Ajuste de diseño
fig.tight_layout()

# Mostrar el gráfico
plt.show()