Обращение из функции к переменной

import math
 
 
# Программа считает время истечения воды из бака
def Re(v, d, nu_=1111.1 * (10 ** (-6))):
    return v * d / nu_
 
 
def lamb(v, d, delta_e=0.05 * 0.001):
    epsilon = delta_e / d
    if Re(v, d) < 2040.0:
        return 64.0 / Re(v, d)
    elif 2040.0 <= Re(v, d) <= 2800.0:
        return (0.16 * Re(v, d) - 13.0) * (10 ** (-4))
    elif 2800.0 < Re(v, d) <= 10.0 / epsilon:
        return 0.3164 / (Re(v, d) ** 0.25)
    elif 10.0 / epsilon < Re(v, d) < 560.0 / epsilon:
        return 0.11 * (epsilon + 68.0 / Re(v, d)) ** 0.25
    else:
        return 0.11 * (epsilon ** 0.25)
 
 
def rs(f, h, a=0.001, b=15, dx=0.001):
    x1 = a
    f1 = f(a, h)
    x2 = a + dx
    f2 = f(x2, h)
    while f1 * f2 > 0.0:
        if x1 >= b:
            return None
        x1 = x2
        f1 = f2
        x2 = x1 + dx
        f2 = f(x2, h)
    return (x1 + x2) / 2
 
 
def func(v, h, d=0.010, Pa=100000, L=1, rho=1262, gg=9.81):
    ksi = 0.82
    return Pa + rho * gg * h + 0.0002 * 0.0002 * 0.5 * rho * v * v - \
           0.5 * rho * v * v - 0.5 * rho * v * v * L * lamb(v, d) / d - ksi * rho * v * v * 0.5
 
 
a = 0.15  # ширина 1
b = 0.15  # ширина 2
D = 0.004
k = math.pi * D * D / (a * b)
H0 = 0.1
n = 20
step = H0 / n
height = [round(H0 - i * step, 3) for i in range(n)]
for i in height:
    print(f'{i} {round(rs(func, i), 3)} {round(k * rs(func, i), 6)} {round(step / (k * rs(func, i)), 5)}')