numpy normalize

# Foe 1d array
an_array = np.array([0.1,0.2,0.3,0.4,0.5])

norm = np.linalg.norm(an_array)
normal_array = an_array/norm
print(normal_array)

#[0.2,0.4,0.6,0.8,1] (Should be, I didin't run the code)

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