Как задать модель уравнением

Есть уравнение профиля:

N1 = N2*math.exp(1/2*(1-(z-z2)/dz2-math.exp(-(z-z2)/dz2))) где z,z2,dz2 это координаты z вычисляется из аппроксимации z2,dz2 задаётся

Потом добавляю ещё одно уравнение:

drv = math.exp(-math.pow((r-rd)/sigma,2)) r = x rd = rd-(z*math.tan(0*math.pi/180))

выходит такой рисунок, итоговое уравнение получается таким

N = N1*(1-drv)

но надо закрыть туннель сверху, каким уравнением это сделать? или добавить 3-фрагмент сверху?