将类内的功能应用于熊猫数据框

时间:2019-05-23 13:09:00

标签: python pandas class

我有一个带有两个函数的类

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一起应用:但不确定如何从类中调用该函数。可能有更好的方法。

0 个答案:

没有答案