import numpy as np c_d = 0.47 # drag coefficient balloon (spherical) [-] def drag(c_d, rho_air, d_b, v): return 0.125 * np.pi * c_d * rho_air * (d_b * v) ** 2