Início / coding / Perguntas / 78314515
Accepted
AmirX
Asked: 2024-04-12 14:24:34 +0800 CST

Qual é a melhor maneira de dividir um dataframe incluindo a primeira instância de uma máscara?

  • 772

Este é o meu DataFrame:

import pandas as pd
import numpy as np
df = pd.DataFrame(
    {
        'a': [np.nan, np.nan, np.nan, 20, 12, 42, 33, 32, 31],
        'b': [np.nan, np.nan, np.nan, np.nan, 2333, np.nan, np.nan, 12323, np.nan]
    }
)

Máscara é:

mask = (
    (df.a.notna()) &
    (df.b.notna())
)

Resultado esperado: fatiar dfaté a primeira instância do mask. Observe que a primeira linha do maskestá INCLUÍDA:

      a        b
0   NaN      NaN
1   NaN      NaN
2   NaN      NaN
3  20.0      NaN
4  12.0   2333.0

Esta primeira instância maské row 4. Portanto, dividi-lo nesse índice é o objetivo.

Estas são minhas tentativas. O primeiro funciona, mas não tenho certeza se a abordagem está correta:

# attempt 1
idx = df.loc[mask.cumsum().eq(1) & mask].index[0]
df = df.loc[:idx]
print(df)
# attempt 2
out = df[~mask.cummax()]
python

1 respostas

  1. Best Answer

    Adicione DataFrame.shiftà sua segunda solução:

    out = df[~mask.shift(fill_value=False).cummax()]
print (out)
      a       b
0   NaN     NaN
1   NaN     NaN
2   NaN     NaN
3  20.0     NaN
4  12.0  2333.0
    • 3

relate perguntas