我需要计算一条线和水平线之间的角度。我的高中数学似乎让我失望。
import matplotlib.pyplot as plt
import numpy as np
x = [8450.0, 8061.0, 7524.0, 7180.0, 8247.0, 8929.0, 8896.0, 9736.0, 9658.0, 9592.0]
y = range(len(x))
best_fit_line = np.poly1d(np.polyfit(y, x, 1))(y)
slope = (y[-1] - y[0]) / (x[-1] - x[0])
angle = np.arctan(slope)
print 'slope: ' + str(slope)
print 'angle: ' + str(angle)
plt.figure(figsize=(8,6))
plt.plot(x)
plt.plot(best_fit_line, '--', color='r')
plt.show()
结果如下:
slope: 0.00788091068301
angle: 0.00788074753125
我需要水平和红色虚线之间的角度。只要看一下它就应该在30-45度之间。我做错了什么?
关于slope = (y[-1] - y[0]) / (x[-1] - x[0])
的 *,我也尝试了numpy.diff
和scipy.stats.linregress
,但也没有成功。
答案 0 :(得分:15)
该线在x方向上从0到9,在y方向从7500到9500.因此,slope
仅为0.00788091068301而不是0.57,约为30°。您的计算是正确的,但最好使用arctan2:
angle = np.rad2deg(np.arctan2(y[-1] - y[0], x[-1] - x[0]))