import matplotlib.pyplot as plt

# Datos
meses = list(range(1, 13))
potencia_termica_cuernavaca = [19603, 26761, 28206, 24357, 28611, 26693, 31247, 26862, 23107, 18040, 22186, 20346]
potencia_termica_mexicali = [18603, 24969, 28971, 31844, 33451, 32520, 31819, 29755, 28131, 24024, 19767, 16647]

# Dividir los valores por 1000 para convertir de kW a MW
potencia_termica_cuernavaca = [p/1000 for p in potencia_termica_cuernavaca]
potencia_termica_mexicali = [p/1000 for p in potencia_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 potencia térmica de Cuernavaca en color celeste
plt.plot(meses, potencia_termica_cuernavaca, marker='o', color='deepskyblue', markersize=8, linestyle='-', linewidth=2, label='Potencia térmica de Cuernavaca')

# Graficar potencia térmica de Mexicali en color morado
plt.plot(meses, potencia_termica_mexicali, marker='s', color='purple', markersize=8, linestyle='-', linewidth=2, label='Potencia térmica de Mexicali')

# Configuraciones adicionales del gráfico
plt.xlabel('Mes', fontsize=22, fontname='Times New Roman')
plt.ylabel('Potencia (kW)', fontsize=22, fontname='Times New Roman')
plt.title('Comparación de potencias térmicas', 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 entre 15 y 35 para una mejor visualización
plt.ylim(15, 37)
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_potencias_termicas.png', dpi=100)

# Mostrar el gráfico
plt.show()