Início / coding / Perguntas / 78285263
Accepted
newdeveloper
Asked: 2024-04-07 01:20:31 +0800 CST

Pandas agrupam por resultados de filtro

  • 772

Estou usando pandas e obtendo dados e usando group by. No entanto, não tenho certeza de como filtrar dados indesejados de serem retornados. Eu só quero a contagem de 1 e não de 0.

print(df.groupby(item)['event'].apply(lambda x: (x==1).sum()).reset_index(name='count'))
hpb   count
0     55
1     36

Também usei o seguinte e ele está retornando as mesmas informações para todas as colunas.

print(patients.query("event==1 and hbp==1").groupby('hbp').count())

       age    cp    eject  ......
hbp
 1      36     36    36

Minha saída desejada está abaixo.

hpb   count
1     36

Aqui estão meus dados que estou lendo nos pacientes do dataframe. Estou tentando obter os dados do hbp onde o evento é 1 e o hbp é 1 usando consulta ou grupo por. Atualmente estou recebendo as duas 1contagens, 0mas só exibo as contagens de HBP 1. veja a saída desejada acima.

hbp event ..etc
1       1
0       1
1       1
1       1
0       1
pandas

1 respostas

  1. Best Answer

    Sua solução é possível alterar especificando a coluna depois groupbye usando sumou countou size, porque processando 1valores:

    print(patients.query("event==1 and hbp==1")
              .groupby('hbp')['event']
              .sum()
              .reset_index(name='count'))

    Ou primeiro filtre e depois conte hbpos valores por Series.value_counts:

    print (patients.query("event==1 and hbp==1")['hbp']
               .value_counts()
               .reset_index(name='count'))

   hbp  count
0    1     36
    • 0

relate perguntas