Início / coding / Perguntas / 77790846
Accepted
x_Amir_x
Asked: 2024-01-10 11:56:10 +0800 CST

Encontrar o valor máximo entre duas colunas onde uma delas é deslocada

  • 772

Meu DataFrame é:

import pandas as pd
df = pd.DataFrame(
    {
       'a': [20, 9, 31, 40],
       'b': [1, 10, 17, 30],
    }
)

Resultado esperado: Criando colunac

    a   b   c
0  20   1  20
1   9  10  20
2  31  17  17
3  40  30  31

Passos:

cé o valor máximo entre df.be df.a.shift(1).bfill().

insira a descrição da imagem aqui

Minha tentativa:

df['temp'] = df.a.shift(1).bfill()
df['c'] = df[['temp', 'b']].max(axis=1)

É a maneira mais limpa/melhor abordagem?

python

4 respostas

  1. Tente assim

    import pandas as pd

df = pd.DataFrame(
    {
       'a': [20, 9, 31, 40],
       'b': [1, 10, 17, 30],
    }
)
df['c'] = df['a'].shift(1).bfill().combine(df['b'], max)
print(df)

    Saída:

        a   b     c
0  20   1  20.0
1   9  10  20.0
2  31  17  17.0
3  40  30  31.0
    • 1
  2. Best Answer

    Se você não quiser a coluna temporária, poderá substituir os valores na coluna deslocada usando where()uma linha.

    df['c'] = df['a'].shift(1).bfill().where(lambda x: x>df['b'], df['b'])

    Isso é semelhante ao combine()método postado na outra resposta, mas este faz uma comparação vetorizada enquanto combine()faz isso elemento a elemento, então deve ser muito mais rápido à medida que o comprimento do dataframe aumenta.

    • 1

  3. use numpy fmaxde maneira rápida e simples

    import numpy as np
df['c'] = np.fmax(df['a'].shift().bfill(), df['b'])

    df

        a   b     c
0  20   1  20.0
1   9  10  20.0
2  31  17  17.0
3  40  30  31.0
    • 1

  4. ImportError: não foi possível alterar o nome 'DataError' de 'pandas.core.base' (/Users/ernidiahsusanti/anaconda3/lib/python3.11/site-packages/pandas/core/base.py)

    • -1

relate perguntas