Я хотел бы иметь норму одного массива NumPy. Более конкретно, я ищу эквивалентную версию этой функции

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

Есть ли что-то подобное вskearn или же ?numpy

Эта функция работает в ситуации, когдаv это вектор 0.