Encontrando valores máximos em várias colunas no Pandas

Para determinar os valores máximos em várias colunas em um DataFrame do pandas, várias abordagens podem ser empregadas . Veja como você pode conseguir isso:

Usando a função max() com colunas especificadas

Este método envolve selecionar explicitamente as colunas desejadas e aplicar a função max():

df[["A", "B"]]
df[["A", "B"]].max(axis=1)

Isso criará uma nova coluna com os valores máximos das colunas A e B.

Usando a função max() com todas as colunas

Se tiver certeza de que o DataFrame contém apenas as colunas para as quais deseja encontrar o máximo, você pode usar a seguinte sintaxe simplificada:

df.max(axis=1)

Isso considerará automaticamente todas as colunas e gerará uma coluna com os valores máximos.

Usando a função apply()

Como alternativa, você pode utilizar a função apply() com a função max:

df.apply(max, axis=1)

Isso irá crie também uma coluna com os valores máximos para cada linha.

Exemplo:

Vamos ilustrar essas abordagens com um exemplo:

import pandas as pd

df = pd.DataFrame({"A": [1, 2, 3], "B": [-2, 8, 1]})

# Using max() with specified columns
df["C"] = df[["A", "B"]].max(axis=1)

# Using max() with all columns
df["D"] = df.max(axis=1)

# Using apply()
df["E"] = df.apply(max, axis=1)

print(df)

Saída:

   A  B  C  D  E
0  1 -2  1  1  1
1  2  8  8  8  8
2  3  1  3  3  3