import matplotlib.pyplot as plt

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

energia_electrica = [
    0.784269, 0.58954, 0.165323, 0.51315, 1.00223, 
    1.87998, 1.298032, 0.213621, 0.18018, 0.452786, 
    0.70155, 0.730949
]

energia_termica = [
    3.824222, 4.560164, 7.203594, 8.70813, 9.881529, 
    11.11749, 10.384287, 8.098595, 7.79763, 6.074419, 
    4.3938, 2.672603
]

# 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 GWh

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