Numpy 3d Distance Function