import matplotlib.pyplot as plt

# Datos de los meses y energía producida en MWh
meses = ['Enero', 'Febrero', 'Marzo', 'Abril', 'Mayo', 'Junio', 
         'Julio', 'Agosto', 'Septiembre', 'Octubre', 'Noviembre', 'Diciembre']

# Energía producida en Cuernavaca y Mexicali
energia_cuernavaca = [
    4.82, 4.92, 6.65, 5.77, 6.19, 
    6.19, 6.97, 4.72, 4.72, 5.00, 
    5.75, 4.75
]

energia_mexicali = [
    4.57, 5.10, 7.30, 9.13, 10.78, 
    12.89, 11.58, 8.23, 7.90, 6.47, 
    5.05, 3.38
]

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

# Graficar energía de Cuernavaca
ax.plot(meses, energia_cuernavaca, marker='o', color='blue', linestyle='-', linewidth=2, label='Localidad de Cuernavaca (MWh)')
# Graficar energía de Mexicali
ax.plot(meses, energia_mexicali, marker='s', color='red', linestyle='-', linewidth=2, label='Localidad de Mexicali (MWh)')

# Configuración del eje Y
ax.set_ylabel('Energía total ahorrada (MWh)', 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(energia_cuernavaca), max(energia_mexicali)) + 2)  # Aumenta el límite superior para MWh

# Título general
plt.title('Energía total ahorrada 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()