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