如何找到趋势线并计算 X 轴趋势线的斜率

Question

我有一个如下所示的 Pandas 数据框：

 UNDERLAY   TIME
 27,395     09:15:18
 27,466     09:17:19
 27,391     09:19:06
 27,409     09:21:19
 27,439     09:23:21
 27,468     09:25:58
 27,497     09:27:19
 27,502     09:29:54
 27,542     09:31:19
 27,522     09:33:33
 27,520     09:35:09
 ...

我想绘制这些UNDERLAY值的趋势线并计算 X 轴的斜率。

从下面的链接中得到了一些帮助，但找不到斜率： 如何在 matplot 上绘制散点趋势线？Python-Pandas

Answer 1

seanborn.regplot 是制作情节的最快方法：

import seaborn as sns

df_plot = pd.DataFrame()
# seconds since midnight of each TIME value
df_plot['SECONDS'] = (pd.to_datetime(df['TIME']) - pd.Timestamp.now().normalize()).dt.total_seconds()
df_plot['UNDERLAY'] = pd.to_numeric(df['UNDERLAY'].str.replace(',', ''))

ax = sns.regplot(data=df_plot, x='SECONDS', y='UNDERLAY')
ax.set(
    xticklabels=pd.to_datetime(ax.get_xticks(), unit='s').strftime('%H:%M'),
    xlabel='Time',
    ylabel='Underlay'
)
plt.show()

输出：

要获得回归函数，请使用 numpy：

import numpy as np
f = np.polyfit(df_plot['SECONDS'], df_plot['UNDERLAY'], deg=1)

# Slope
f[0]

# Make a prediction at 21:00
# Time is expressed as seconds since midnight
np.polyval(f, 21*3600)