def dist(x1, x2, axis=0):
    """Return the distance between two points.

    Set axis=1 if x1 is a vector and x2 a matrix to get a vector of distances.
    """
    return np.linalg.norm(x2 - x1, axis=axis)
Return the distance between two points.

    Set axis=1 if x1 is a vector and x2 a matrix to get a vector of distances.