我有一个带有两个函数的类
import pandas as pd
class fees():
def __init__(self):
self.total_fee = 0
def base_fee(self, value):
if value <= 200:
fee = value * 0.1
self.total_fee += fee
return fee
elif value > 200:
fee = value * 0.07
self.total_fee += fee
return fee
def com_fee(self, value):
if value <= 10:
fee = value*0.2
self.total_fee += fee
return fee
elif value > 10:
fee = value *0.1
self.total_fee += fee
return fee
这将返回in内的总费用。但是,我也试图使用该类中的函数在下面的数据框中添加两列。
df = pd.DataFrame({
'person': ['a', 'b', 'c'],
'total_spend': [1000, 150, 45],
'additional_spend': [25, 5, 0]
})
结果将是
df1 = pd.DataFrame({
'person': ['a', 'b', 'c'],
'total_spend': [1000, 150, 45],
'additional_spend': [25, 5, 0],
'base_fee': [70, 15, 4.5],
'com_fee': [7, 1.5, 0]
})
我唯一尝试过的就是将函数与lambda x一起应用:但不确定如何从类中调用该函数。可能有更好的方法。