Function unit_vector

src/bcf/bcf/geometry.py:44–55

Return the unit vector of a vector. Args: v: vector Returns: unit vector.

(v: NDArray[np.float64])

def unit_vector(v: NDArray[np.float64]) -> NDArray[np.float64]:
    """
    Return the unit vector of a vector.

    Args:
        v: vector

    Returns:
        unit vector.
    """
    norm = np.linalg.norm(v)
    return v if norm == 0 else v / norm