如何解决如何向该数据框添加趋势线Python
我想知道如何在条形图中添加趋势线(或最佳拟合线)。我尝试搜索,但发现自己很困惑。这是我用来创建条形图的代码:
dfY = pd.DataFrame(data={"Year":dataYVote['Year'],"Vote": dataYVote['Vote']})
year_list = []
avg_votes = []
for year in np.unique(dfY["Year"]):
year_list.append(year)
avg_votes.append(dfY.loc[dfY["Year"]==year,"Vote"].mean())
plt.bar(year_list,avg_votes,width = 0.5)
plt.xlabel('Year')
plt.ylabel('Amount of Votes')
plt.title('Average Amount of Votes for Each Year')
plt.show()
非常感谢您的帮助:)
解决方法
您可以使用numpy.polyfit
来使平方误差最小,并返回梯度和截距。
import numpy as np
slope,intercept = np.polyfit(x_data,y_data,1) #deg of 1 for straight line
plt.plot(x,slope*x + intercept)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。