Нахождение максимальных значений по нескольким столбцам в Pandas

Чтобы определить максимальные значения по нескольким столбцам в Pandas DataFrame, можно использовать различные подходы . Вот как этого можно добиться:

Использование функции max() с указанными столбцами

Этот метод включает в себя явный выбор нужных столбцов и применение функции max():

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

Это создаст новый столбец с максимальными значениями из столбцов A и B.

Использование функции max() со всеми столбцами

Если вы уверены, что DataFrame содержит только те столбцы, для которых вы хотите найти максимум, вы можете использовать следующий упрощенный синтаксис:

df.max(axis=1)

Это автоматически учтет все столбцы и выведет столбец с максимальными значениями.

Использование функции apply()

В качестве альтернативы вы можете использовать функцию apply() с функцией max:

df.apply(max, axis=1)

Это будет также создайте столбец с максимальными значениями для каждой строки.

Пример:

Давайте проиллюстрируем эти подходы на примере:

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)

Вывод:

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