friction.py

import numpy as np

def truncated_friction(params, v, alpha):
   vmin = params['V0'] / np.exp( ( params['mu0'] + params['b'] * np.array(alpha)) / params['a'] )
   clipped_v = (v/vmin).clip(1)
   return params['a'] * np.log(clipped_v)

   #return (params['a'] * np.log(v/params['V0']) - params['mu0'] - (params['b']/params['a'])*np.array(alpha) ).clip(0)