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']

energia_electrica = [
    0.740249, 0.607824, 0.402721, 0.54759, 0.521916, 
    0.4482, 0.285231, 0.720068, 0.69744, 0.71548, 
    0.58524, 0.750138
]

energia_termica = [
    4.117637, 4.359264, 6.306609, 5.27748, 5.729203, 
    5.80422, 6.751986, 4.038525, 4.06269, 4.323012, 
    5.21604, 4.043082
]

# 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 eléctrica
ax.plot(meses, energia_electrica, marker='o', color='blue', linestyle='-', linewidth=2, label='Energía Eléctrica (MWh)')
# Graficar energía térmica
ax.plot(meses, energia_termica, marker='s', color='red', linestyle='-', linewidth=2, label='Energía Térmica (MWh)')

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

# Título general
plt.title('Energía producida por mes para la localidad de Cuernavaca', 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()