How To Get The Unit Vector From A Numpy Array

Lets say I have a vector v, and I want the unit vector, i.e. v has length 1.0 Is there a direct way to get that from numpy? I want something like: import numpy as np v=np.arrange(3

Solution 1:

There's no function in numpy for that. Just divide the vector by its length.

v_hat = v / (v**2).sum()**0.5

or

v_hat = v / np.linalg.norm(v)