apply python function pandas code example

Example 1: 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)

Example 2: pandas apply function to dataframe

# iterate using this syntax where df is the pandas.DataFrame
# N. B.: a single tuple is composed by the row index 
# and the values of the dataframe columns  

for row in df.itertuples():
  # do your stuff here