我想编写一个函数,它接受两个长度相同的 numpy 数组并返回:
两个数组的点积
两个向量之间的角度(以度为单位)
预期输出如下:
print(angle_dot(np.array([0., 1.]), np.array([1., 0.])))
## (0.0, 90.0)
print(angle_dot(np.array([2., -1, 1, -2]), np.array([-1., 1.5, 3., 1])))
## (-2.5, 102.5)
这是我到目前为止所拥有的,我似乎无法获得正确的数组之间的角度值。
import numpy as np
def angle_dot(a, b):
dot_product = round(np.dot(a, b), 1)
angle = round(np.degrees(dot_product), 1)
return dot_product, angle
print(angle_dot(np.array([0., 1.]), np.array([1., 0.])))
## (0.0, 0.0)
print(angle_dot(np.array([2., -1, 1, -2]), np.array([-1., 1.5, 3., 1])))
## (-2.5, -143.2)
收到一只叮咚
炎炎设计
相关分类