how to normalize a vector in python code example

Example: numpy normalize

def normalize(v):
    norm = np.linalg.norm(v)
    if norm == 0: 
       return v
    return v / norm