在编写一个函数来计算单位向量时,需要将向量的每个分量除以向量的模长。下面是一个示例代码:
import math
def unit_vector(vector):
magnitude = math.sqrt(sum(component**2 for component in vector))
if magnitude == 0:
return vector
return [component / magnitude for component in vector]
# 测试代码
vector = [3, 4] # 示例向量
unit_vector = unit_vector(vector)
print(unit_vector)
在上面的代码中,unit_vector
函数接受一个向量作为输入参数,并计算该向量的模长。然后,通过将向量的每个分量除以模长,得到单位向量。如果向量的模长为零,则返回原始向量。
在测试代码中,我们使用了一个示例向量 [3, 4]
,然后调用 unit_vector
函数来计算单位向量。最后,打印结果 [0.6, 0.8]
。这表示 [3, 4]
的单位向量是 [0.6, 0.8]
。
上一篇:编写一个函数来计算乘积,例如