Euclidean Distance 3d Python