apply_pandas.py

__author__ = "MUTHUPANDIAN"
__email__ = "contact@muthupandian.in"

'''
PERFORM SOME OPERATIONS IN THE EXISTING DATAFRAME USING APPLY METHOD
''''

import pandas as pd

#FUNCTIONS TO PERFORM OPERATION BETWEEN TWO COLUMNS
def sum(cols):
    return cols[0]+cols[1]

def avg(cols):
    return (cols[0]+cols[1])/2

def mul(cols):
    return cols[0]*cols[1]


#CREATE NEW DUMMY DATAFRAME
df = pd.DataFrame({'a':[10,20,30],'b':[33,44,55]})
print(df)

#APPENDING THE COLUMNS TO THE EXISTING DATAFRAME WITH THE OUTPUT FROM FUNCTIONS
df['total']=(df.apply(sum,axis=1))
df['average']=(df.apply(avg,axis=1))
df['multiplication']=(df.apply(mul,axis=1))

#PRINTING DATAFRAME AFTER ADDING THE NEW COLUMNS
print(df)

new commit