I would like to have the norm of one NumPy array. More specifically, I am looking for an equivalent version of this function

def normalize(v):

norm = np.linalg.norm(v)

if norm == 0:

return v

return v / norm

Is there something like that in sklearn or numpy?

This function works in a situation where v is the 0 vector.