numpy 为时间序列图添加趋势线

6ju8rftf  于 2023-05-29  发布在  其他
关注(0)|答案(2)|浏览(191)

我想在python中为一个时间序列图添加一个趋势线,这意味着我的x轴(基准)的格式为datetime64[ns],当我跟随这个线程时:How to add trendline in python matplotlib dot (scatter) graphs?
运行我的代码:

import numpy as np
#Trendlines
z = np.polyfit(df1['Datum'], df1['Score'], 1)
p = np.poly1d(z)

我得到错误:
UFuncTypeError:ufunc“add”不能使用dtype('类型的操作数
我该如何解决这个问题?这个thread也没有帮助

tgabmvqs

tgabmvqs1#

解决方法是:

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
x = mdates.date2num(df1['Datum'])
y= df1['Score']
z = np.polyfit(x, df1['Score'], 1)
p = np.poly1d(z)
#then the plot
df1.plot('Datum', 'Score')
plt.plot(x, p(x), "r--")

给出带有折线图和趋势线的结果

8yoxcaq7

8yoxcaq72#

#  date         value
# 01-01-2021    141937
# 01-02-2021    41204
# 01-03-2021    2198991
# 01-04-2021    8744873
# 01-09-2021    7825446

x = df['date']
x = pd.to_datetime(x)
plt.scatter(x, y)

y = df['value'].tolist()
x = dates.date2num(list(pd.to_datetime(x)))
z = np.polyfit(x,y,1)
p = np.poly1d(z)
plt.plot(pd.to_datetime(a['date']), p(x), "r--")

相关问题