def get_r( x, y = 0, z = 1):
    """Makes r (length between origin and point) for x, y, z

    Returns r (scalar or array, depending on inputs)"""

    r = ( x **2 + y **2 + z **2 ) **0.5

    return r