Início / coding / Perguntas / 78566699
Accepted
Giampaolo Levorato
Asked: 2024-06-02 22:20:32 +0800 CST

Obtenha o primeiro elemento diferente de zero em pandas groupby

  • 772

Eu criei o seguinte dataframe do pandas:

import pandas as pd

ds = {'cycle' : [1,1,1,1,2,2,2,2], 'values' : [0,0,-1,1,2,0,0,1]}

df = pd.DataFrame(data=ds)

O quadro de dados fica assim:

df

  cycle values
0   1   0
1   1   0
2   1   -1
3   1   1
4   2   2
5   2   0
6   2   0
7   2   1

Preciso criar um novo dataframe (chamado ds) que contenha apenas o primeiro registro com valor diferente de zero para as colunas valuesde cada um cycle.

Eu acho que isso pode ser feito com algo em torno do df.groupby(['cycle'])código.

Portanto, o dataframe resultante ficaria assim:

   cycle    values
0   1        -1
1   2         2
pandas

2 respostas

  1. Remova os itens nulos (por exemplo query, com ) e então drop_duplicates:

    df.query('values != 0').drop_duplicates(subset='cycle')

    Saída:

       cycle  values
2      1      -1
4      2       2
    • 2
  2. Best Answer

    Você pode

    • filterfora os elementos zero

    • use groupbypara agrupar por ciclo

    • extrair o firstelemento para cada grupo

      df[df['values'] != 0].groupby(['cycle']).first()
    • 1

relate perguntas