Recherche de valeurs maximales sur plusieurs colonnes dans Pandas

Pour déterminer les valeurs maximales sur plusieurs colonnes dans un DataFrame pandas, diverses approches peuvent être utilisées . Voici comment y parvenir :

Utilisation de la fonction max() avec les colonnes spécifiées

Cette méthode implique de sélectionner explicitement les colonnes souhaitées et d'appliquer la fonction max() :

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

Cela créera une nouvelle colonne avec les valeurs maximales des colonnes A et B.

Utilisation de la fonction max() avec toutes les colonnes

Si vous êtes sûr que le DataFrame contient uniquement les colonnes pour lesquelles vous souhaitez trouver le maximum, vous pouvez utiliser la syntaxe simplifiée suivante :

df.max(axis=1)

Cela prendra automatiquement en compte toutes les colonnes et affichera une colonne avec les valeurs maximales.

Utilisation de la fonction apply()

Vous pouvez également utiliser la fonction apply() avec la fonction max :

df.apply(max, axis=1)

Cela permettra créez également une colonne avec les valeurs maximales pour chaque ligne.

Exemple :

Illustrons ces approches avec un exemple :

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)

Sortie :

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