Cursos repetidos por aluno | Pandas: formatos diferentes de entrada e saída (IO)

Solucionado (ver solução)

Solucionado
(ver solução)

3
respostas

por LUIS FELIPE TOMINAGA DIELLE

| 73.5k xp | 29 posts

Olá, tudo bem?

No exemplo da aula, os cursos são atribuídos aos alunos conforme o número de matrículas. Porém existem alunos que estão matriculados mais de uma vez no mesmo curso. Utilizei o drop_duplicates() mas mesmo assim parece não funcionar.

for index, row in nomes.iterrows(): 
    id = row.id_aluno
    matriculas = row.matriculas 
    for i in range(matriculas): 
        mat = [id, np.random.choice(cursos.index, p = prob)] 
        qtd_matriculas.append(mat) 

matriculas = pd.DataFrame(qtd_matriculas, columns = ['id_aluno', 'id_curso'])

matriculas = matriculas.drop_duplicates()

Como resolver? Obrigado!

3 respostas

solução!

por Cleiton Rodrigues da Silva

| 65.7k xp | 33 posts

15/07/2020

Luiz,

Por favor poderia tentar com o trecho de código abaixo:

matriculas.drop_duplicates(subset =['id_aluno', 'id_curso'], keep = False, inplace = True)

Fonte: https://www.geeksforgeeks.org/python-pandas-dataframe-drop_duplicates/

por LUIS FELIPE TOMINAGA DIELLE

| 73.5k xp | 29 posts

15/07/2020

Cleiton, obrigado pela ajuda.

Li o link que você enviou e apenas alterei uma informação. No keep ao invés de colocar False coloquei 'first', pois com o False ele apaga não só os valores duplicados e sim todos os valores dentro dos parâmetros.

Aproveitei também em fiz um check de quantos registros existiam antes e depois do drop_duplicate.

Segue abaixo o código.

l1 = len(matriculas)
l1

matriculas.drop_duplicates(subset =['id_aluno', 'id_curso'], keep = 'first', inplace = True)

l2 = len(matriculas)
l2

Muito obrigado pela ajuda!

por Constantin Coucolis Junior

| 42.5k xp | 18 posts

29/07/2020

Ia fazer essa pergunta também, o código sugerido pelo Cleiton e editado pelo Luis Felipe resolveu o caso.