Say I want to iterate say I have pandas dataframe:
import pandas as pd
tmp = pd.DataFrame([(2,2),(0,4),(1,3)],columns= ['A','B'])
and I want to write something of the following kind:
def f(operator,column)
s.t
f('mean','A')
will return:
tmp['A'].mean()
IIUC, use getattr
:
tmp = pd.DataFrame([(2,2),(0,4),(1,3)], columns=['A','B'])
def f(operator, column):
return getattr(tmp[column], operator)()
f('mean', 'A')
# 1.0
f('max', 'B')
# 4
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With