import matplotlib.pyplot as plt

# Datos
meses = list(range(1, 13))
potencia_cuernavaca = [8313, 10975, 11262, 9936, 11606, 10826, 12381, 11012, 9737, 7758, 9337, 8694]
potencia_mexicali = [8273, 10486, 11618, 12372, 12889, 12433, 12306, 11688, 11237, 10128, 8746, 7533]

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

# Graficar potencia eléctrica de Mexicali en color morado
plt.plot(meses, potencia_mexicali, marker='s', color='purple', markersize=8, linestyle='-', linewidth=2, label='Potencia eléctrica 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 eléctricas', 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 7 y 14 para una mejor visualización
plt.ylim(7, 14)
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_electricas.png', dpi=100)

# Mostrar el gráfico
plt.show()