Note that for that calculation, hypot would probably be even better (because of numerical precision issues if nothing else):
u = np.hypot(location.x, location.y).to('m').value