python numpy生成单位向量

可以使用numpy的linalg模块中的norm函数来计算向量的模长，然后将向量除以模长即可得到单位向量。具体实现如下：

import numpy as np

def unit_vector(vector):
    """返回向量的单位向量"""
    return vector / np.linalg.norm(vector)

# 示例
v = np.array([1, 2, 3])
u = unit_vector(v)
print(u)  # 输出 [0.26726124 0.53452248 0.80178373]