apply function to a dataframe code example

Example 1: pandas apply function to column

df['a'] = df['a'].apply(lambda x: x + 1)

Example 2: give function to pandas apply

import pandas as pd


def sum(x, y, z, m):
    return (x + y + z) * m


df = pd.DataFrame({'A': [1, 2], 'B': [10, 20]})

df1 = df.apply(sum, args=(1, 2), m=10)
print(df1)