Início / coding / Perguntas / 77791091
Accepted
x_Amir_x
Asked: 2024-01-10 13:25:51 +0800 CST

Encontrar o valor máximo entre duas colunas onde uma delas é deslocada e alterar o valor da última linha

  • 772

Meu DataFrame é:

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

Resultado esperado: Criando coluna cename

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

Passos:

a) cé criado pordf['c'] = np.fmax(df['a'].shift().bfill(), df['b'])

b) para a última linha: df['c'] = df[['a', 'b']].max(). Já para a última linha a > b40 é escolhida.

c) Obtenha o nome do valor máximo entre aou bpara a última linha.

Minha tentativa:

df['c'] = np.fmax(df['a'].shift().bfill(), df['b'])
df.loc[df.index[-1], 'c'] = df.loc[df.index[-1], ['a', 'b']].max()
df.loc[df.index[-1], 'name'] = df.loc[df.index[-1], ['a', 'b']].idxmax()

É a maneira mais limpa/melhor abordagem?

python

1 respostas

  1. Best Answer

    Não sei o quanto isso representa uma melhoria, mas você pode combinar as duas últimas linhas de código em uma única linha se usar o agg().

    df['c'] = np.fmax(df['a'].shift().bfill(), df['b'])
idx = df.index[-1]
df.loc[idx, ['c', 'name']] = df.loc[idx, ['a', 'b']].agg(['max', 'idxmax']).to_numpy()

    Para criar uma cópia, poderíamos definir uma máscara que sinaliza a última linha e assign()as colunas “c” e “nome”.

    msk = df.index == df.index[-1]
df1 = df.assign(
    c=np.fmax(df['a'].shift().bfill().mask(msk, df['a']), df['b']), 
    name=df[['a', 'b']].idxmax(axis=1).where(msk)
)
    • 1

relate perguntas