adega#46 Listagem dos iras médios

src/script/analysis/admission_analysis.py

 import numpy as np
 
 from script.utils.situations import *
+from script.analysis.student_analysis import ira_alunos
+from collections import defaultdict
+
+import numpy as np
+
 ANO_ATUAL = 2017
 SEMESTRE_ATUAL = 2
 
 
-def listagem_turma_ingresso(df):
-	#~ print(df.groupby(["ANO_INGRESSO", "SEMESTRE_INGRESSO"]).groups)
-	grupos = df.groupby(["ANO_INGRESSO", "SEMESTRE_INGRESSO"]).groups
-	for t in grupos:
-		print(t)
-		print("\n\n")
-		print(df["FORMA_INGRESSO"][grupos[t]].drop_duplicates())
+def media_ira_turma_ingresso(df):
+	iras = ira_alunos(df)
+	
+	turmas_ingresso_grr = df.groupby([
+		"ANO_INGRESSO",
+		"SEMESTRE_INGRESSO",
+		"MATR_ALUNO"]
+	).groups
+	
+	# Cria um dicionario cujas chaves são GRR
+	# e valor são tuplas (ano_ingresso,semestre_ingresso)
+	ano_semestre_do_grr = {}
+	for ti in turmas_ingresso_grr:
+		ano_semestre_do_grr[ ti[2] ] = (ti[0],ti[1])
+	
+	resultados = defaultdict(list)
+	
+
+	for grr in iras:
+		semestre_ano = ano_semestre_do_grr[grr]
+		resultados[ semestre_ano ].append(iras[grr])
+	
+	
+	# Calcula a média do ira para cada turma_ingresso
+	for r in resultados:
+		aux = np.array(resultados[r])
+		resultados[r] = np.mean(aux)
+	
+	return resultados

src/script/build_cache.py

@@ -19,15 +19,13 @@ def build_cache(dataframe):
 	ensure_path_exists(path)
 
 	for cod, df in dataframe.groupby('COD_CURSO'):
-		path = path + '/' + cod + '/'  
+		path = path + '/' + cod + '/'
 		generate_degree_data(path, df)
 		generate_student_data(path+'students/',df)
-		#~ generate_admission_data(path+'/'+cod+'/admission/',df)
-	#generate_degree_data(path, dataframe)
-	#generate_student_data(path, dataframe)
-	#generate_student_list(path)
-	#generate_admission_data(path)
-	#generate_admission_list(path)
+		generate_admission_data(path+'/admission/',df)
+		#generate_student_list(path)
+		generate_admission_data(path, dataframe)
+		generate_admission_list(path, dataframe)
 		generate_course_data(path+'disciplina/' ,dataframe)
 
 def generate_degree_data(path, dataframe):
@@ -140,10 +138,23 @@ def generate_student_list(path):
 	pass
 
 def generate_admission_data(path,df):
-	listagem_turma_ingresso(df)
-	pass
+	
+	media_ira = media_ira_turma_ingresso(df)
+	listagem = []
+	
+	# x é uma tupla (ano,semestre) 
+	for x in media_ira:
+		listagem.append({
+			"ira": media_ira[x],
+			"ano": x[0],
+			"semestre": x[1]
+		})
+	
+
+	save_json(path+"lista_turma_ingresso.json", listagem)
+	
 
-def generate_admission_list(path):
+def generate_admission_list(path,df):
 	pass
 
 def generate_course_data(path,df):