DataFrame.apply()

Применяет функцию к столбцам или строкам.

СигнатураDataFrame.apply(func, axis=0)

Применяет функцию вдоль оси: axis=0 — к каждому столбцу, axis=1 — к каждой строке. Для Series применяет функцию к каждому элементу.

import pandas as pd
df = pd.DataFrame({'a': [1, 2], 'b': [3, 4]})
print(df.apply(sum, axis=1))
# 0    4
# 1    6