Distance Function Numpy Python