import matplotlib.pyplot as plt

# Datos
meses = list(range(1, 13))
energia_termica_cuernavaca = [6743088, 8911746, 10492632, 9158608, 11529427, 10117026, 12718343, 10610885, 4769532, 3553407, 7477019, 6937986]
energia_termica_mexicali = [5897151, 7940142, 10748241, 12355472, 14317028, 13886040, 13873084, 12288815, 10436601, 8480472, 6226605, 5177217]

# Convertir Wh a MWh
energia_termica_cuernavaca = [x / 1_000_000 for x in energia_termica_cuernavaca]
energia_termica_mexicali = [x / 1_000_000 for x in energia_termica_mexicali]

# Configuración del gráfico
plt.figure(figsize=(10.67, 8))  # Tamaño en pulgadas para una escala 4:3 y resolución 800x600 (10.67 x 8 pulgadas)

# Graficar energía térmica de Cuernavaca en color celeste
plt.plot(meses, energia_termica_cuernavaca, marker='o', color='deepskyblue', markersize=8, linestyle='-', linewidth=2, label='Energía térmica de Cuernavaca')

# Graficar energía térmica de Mexicali en color morado
plt.plot(meses, energia_termica_mexicali, marker='s', color='purple', markersize=8, linestyle='-', linewidth=2, label='Energía térmica de Mexicali')

# Configuraciones adicionales del gráfico
plt.xlabel('Mes', fontsize=22, fontname='Times New Roman')
plt.ylabel('Energía (MWh)', fontsize=22, fontname='Times New Roman')
plt.title('Comparación de energía térmica generada', fontsize=22, fontname='Times New Roman')

# Ajuste del tamaño de fuente para las leyendas dentro del gráfico
plt.legend(loc='upper left', fontsize=16)

# Mostrar marco alrededor de los ejes
plt.gca().spines['top'].set_linewidth(1.5)  # Grosor del borde superior
plt.gca().spines['right'].set_linewidth(1.5)  # Grosor del borde derecho
plt.gca().spines['bottom'].set_linewidth(1.5)  # Grosor del borde inferior
plt.gca().spines['left'].set_linewidth(1.5)  # Grosor del borde izquierdo

# Establecer el rango del eje Y de 0 a 16
plt.ylim(1, 17)
plt.yticks(fontsize=18)

# Establecer los ticks del eje X
plt.xticks(meses, fontsize=18)

# Estilo de la fuente general
plt.rcParams['font.family'] = 'Times New Roman'

# Eliminación de cuadrícula
plt.grid(False)

# Guardar la figura con la resolución especificada
plt.tight_layout()
plt.savefig('comparacion_energia_termica.png', dpi=100)

# Mostrar el gráfico
plt.show()