Início / coding / Perguntas / 79255967
Accepted
Shawn Zhang
Asked: 2024-12-06 03:15:59 +0800 CST

Como criar uma coluna de valor crescente com base em outra coluna com o mesmo valor [duplicado]

  • 772

todos

Eu tenho um quadro de dados como abaixo

df = pd.DataFrame.from_dict({'A':[1,1,2], 'B':[10,20,14], "C":[30,40,20]})

insira a descrição da imagem aqui

Como criar uma nova coluna, por exemplo, newColque depende da coluna A. Se o valor em Apermanecer o mesmo, então newColserá preenchido com um inteiro crescente. Se houver um valor diferente em A, aqui 2, então o valor de newColserá redefinido e começará a aumentar novamente.

Obrigado.

insira a descrição da imagem aqui

python

1 respostas

  1. Best Answer

    Código

    grp = df['A'].ne(df['A'].shift()).cumsum()
df['newCol'] = df['A'].groupby(grp).cumcount().add(1)

    df:

       A   B   C  newCol
0  1  10  30       1
1  1  20  40       2
2  2  14  20       1
    • 2

relate perguntas