python - Seaborn regplot 使用 datetime64 作为 x 轴

Question

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

date         score  
2017-06-04    90
2017-06-03    80
2017-06-02    70

当我尝试这样做时:

sns.regplot(x=date, y=score, data=df)

我遇到了一个错误:

TypeError: reduction operation 'mean' not allowed for this dtype

日期的数据类型是datetime64[ns]，分数列的数据类型是int64。

如何隐藏 date 列以便 regplot 可以工作？

Answer 1

Seaborn 不支持 regplot 中的日期时间，但这里有一个丑陋的 hack:

df = df.sort_values('date')
df['date_f'] = pd.factorize(df['date'])[0] + 1
mapping = dict(zip(df['date_f'], df['date'].dt.date))

ax = sns.regplot('date_f', 'score', data=df)
labels = pd.Series(ax.get_xticks()).map(mapping).fillna('')
ax.set_xticklabels(labels)

产生

这是时间序列回归中使用的主要方法。如果您有每日数据，则将第 1 天编码为 1，并随着时间的推移增加数字。这假设您有一个固定间隔的时间序列。