Início / coding / Perguntas / 77251464
Accepted
Fresto
Asked: 2023-10-08 04:45:30 +0800 CST

Como mesclar linhas com valores mais próximos no Dataframe

  • 772

Eu tenho um DataFrameassim:

index B
0     1
1     2
2     5
3     6
4     7
5     10

E preciso mesclar as linhas onde a diferença é menor ou igual a 2, selecionar a linha com o valor menor e definir a contagem de mesclagens

O resultado deve ser assim:

index B count
0     1   2
1     5   3
2     10  1

Como isso pode ser resolvido usando pandas?

python

2 respostas

  1. Best Answer

    Muito semelhante a @AndrejKesely , apenas com sintaxe mais curta, usando agregações nomeadas :

    df.groupby(df["B"].diff().gt(2).cumsum(), 
           as_index=False).agg(B=("B", "first"), 
                               count=("B", "count"))

    Saída:

        B  count
0   1      2
1   5      3
2  10      1
    • 7

  2. Use .groupby()onde você faz .diff()seguido por .cumsum():

    out = (
    df.groupby(df["B"].diff().gt(2).cumsum())
    .agg({"B": ("min", "count")})
    .droplevel(0, axis=1)
    .rename(columns={"min": "B"})
    .reset_index(drop=True)
)

print(out)

    Impressões:

        B  count
0   1      2
1   5      3
2  10      1
    • 5

relate perguntas