Untitled

# 2C : Para cada región, gráfico con la matrícula anual según nivel de enseñanza agrupados
import numpy as np
from matplotlib import pyplot as plt

conn = psycopg2.connect(**params)
cursor = conn.cursor()


cursor.execute(""" 
select nom_reg_rbd,agno,cod_ense2,(sum(n_alumno_hom)+sum(n_alumno_muj)+sum(n_alumno_si)) as matriculados from matricula m
inner join establecimiento e on m.rbd = e.rbd
inner join comuna c on c.cod_com_rbd = e.cod_com_rbd
inner join provincia p on c.cod_pro_rbd = p.cod_pro_rbd
inner join region r on p.cod_reg_rbd = r.cod_reg_rbd
group by 1,2,3 order by cod_ense2 asc;
    """)

rows = cursor.fetchall()

cod_ense2=range(1,9)
agno=range(1,7)

cod={}
cod_a=[]
for r in rows:
  if r[2] not in cod:
    cod[r[2]] = np.zeros((6), dtype=int)
    cod_a.append(r[2])
  cod[r[2]][int(r[1])-2015] = r[3]


fig, ax1 = plt.subplots()
for c in cod_a:
  ax1.plot(agno, cod[c], label=c)

ax1.set_title('Antofa')
ax1.set_xlabel('agnos')
ax1.set_ylabel('matriculas')
ax1.legend()

plt.show()

cursor.close()
conn.close()