如何从numpy数组中获取单位向量

Question

假设我有一个向量v，想要得到单位向量，即v的长度为1.0。是否有一种直接从Numpy获取它的方法？

我想要像这样的结果：

import numpy as np
v=np.arrange(3)
v_hat = v.norm()

与其

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

- CodingFrog

---

是的，我看到了那一个，但是希望自那以后有更简单的方法。 - CodingFrog

v /= np.linalg.norm(v) 可以简洁地表示将向量v归一化。 - Jussi Nurminen

Answer 1

numpy中没有这样的函数。只需将向量除以它的长度即可。

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

或者

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

- blue_note