Início / coding / Perguntas / 77740923
Accepted
x_Amir_x
Asked: 2024-01-01 12:02:28 +0800 CST

Substituir um valor pelo seu valor anterior em uma coluna se for maior

  • 772

Este é o meu DataFrame:

import pandas as pd
df = pd.DataFrame(
    {
        'a': [101, 90, 11, 120, 1]
    }
)

E esta é a saída que eu quero. Quero criar uma coluna y:

     a    y
0  101    101.0
1   90    101.0
2   11    90.0
3  120    120.0
4    1    120.0

Basicamente, os valores in asão comparados com o valor anterior, e o maior é selecionado.

Por exemplo, para row 1, 90 é comparado com 101. 101 é maior, por isso é selecionado.

Eu fiz desta maneira:

df['x'] = df.a.shift(1)
df['y'] = df[['a', 'x']].max(axis=1)

Existe um limpador ou algum tipo de maneira integrada de fazer isso?

python

1 respostas

  1. Best Answer

    Você pode usar np.fmaxpara obter os máximos sem criar uma coluna adicional:

    df["y"] = np.fmax(df["a"], df["a"].shift(1))

    Isso produz:

         a      y
0  101  101.0
1   90  101.0
2   11   90.0
3  120  120.0
4    1  120.0

    Costumamos np.fmax()ignorar o NaNcriado durante a mudança df["a"].

    • 3

relate perguntas