data4_pivot['total_duration'] = round(data4_pivot['total_duration'], 0)
def money(subscription_type):
    if subscription_type == 'ultra':
        return data4_pivot['total_duration'] * 6 + data4_pivot['subscription_type'].count() * 199
    if subscription_type == 'free':
        return data4_pivot['total_duration']* 8 + data4_pivot['trip_count'].count() * 50
data4_pivot['money'] = data4_pivot['subscription_type'].apply(money, axis=1)


---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
/tmp/ipykernel_79/156564176.py in <module>
      5     if subscription_type == 'free':
      6         return data4_pivot['total_duration']* 8 + data4_pivot['trip_count'].count() * 50
----> 7 data4_pivot['money'] = data4_pivot['subscription_type'].apply(money, axis=1)

/opt/conda/lib/python3.9/site-packages/pandas/core/series.py in apply(self, func, convert_dtype, args, **kwds)
   4136             else:
   4137                 values = self.astype(object)._values
-> 4138                 mapped = lib.map_infer(values, f, convert=convert_dtype)
   4139 
   4140         if len(mapped) and isinstance(mapped[0], Series):

pandas/_libs/lib.pyx in pandas._libs.lib.map_infer()

/opt/conda/lib/python3.9/site-packages/pandas/core/series.py in f(x)
   4121 
   4122             def f(x):
-> 4123                 return func(x, *args, **kwds)
   4124 
   4125         else:

TypeError: money() got an unexpected keyword argument 'axis'