import numpy as np def distance(a, b): r = np.sum((a - b) ** 2) ** 0.5 return r