import pandas as pd total_cadeiras = 29 QE = 12684 eleicao = pd.read_csv('eleicao.csv', delimiter=';') # agrupando e somando os valores de cada partido/coligacao partidos_votos = eleicao.groupby(['Partido/Coligação']).sum() partidos_votos['QP'] = partidos_votos['Votos']//QE # ir atualizando a quantide de vagas já preenchidas sumQp = partidos_votos['QP'].sum() # enquanto total_cadeiras não for atigindo, continua a calcular a media while(total_cadeiras != sumQp): partidos_votos['Media'] = partidos_votos['Votos']/(partidos_votos['QP']+1) partidos_votos[partidos_votos['Media'] == partidos_votos['Media'].max()]['QP'] +=1 sumQp +=1 print(sumQp)