import matplotlib.pyplot as plt

# Datos
meses = list(range(1, 13))
energia_elec_cuernavaca = [2859672, 3654675, 4189464, 3736312, 4677218, 4103054, 5039067, 4350135, 2013362, 1539120, 3146569, 2964654]
energia_elec_mexicali = [2622541, 3334548, 4310278, 4800336, 5516492, 5308891, 5365416, 4827144, 4168927, 3575184, 2754990, 2342763]

# Convertir Wh a MWh
energia_elec_cuernavaca = [x / 1_000_000 for x in energia_elec_cuernavaca]
energia_elec_mexicali = [x / 1_000_000 for x in energia_elec_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 eléctrica de Cuernavaca en color celeste
plt.plot(meses, energia_elec_cuernavaca, marker='o', color='deepskyblue', markersize=8, linestyle='-', linewidth=2, label='Energía eléctrica de Cuernavaca')

# Graficar energía eléctrica de Mexicali en color morado
plt.plot(meses, energia_elec_mexicali, marker='s', color='purple', markersize=8, linestyle='-', linewidth=2, label='Energía eléctrica 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 eléctrica 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
plt.ylim(0, max(max(energia_elec_cuernavaca), max(energia_elec_mexicali)) + 1)
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_electrica.png', dpi=100)

# Mostrar el gráfico
plt.show()