我目前正在使用Pandas和matplotlib来执行一些数据可视化,我想在我的散点图中添加一条最佳拟合线。
下面是我的代码:
import matplotlib
import matplotlib.pyplot as plt
import pandas as panda
import numpy as np
def PCA_scatter(filename):
matplotlib.style.use('ggplot')
data = panda.read_csv(filename)
data_reduced = data[['2005', '2015']]
data_reduced.plot(kind='scatter', x='2005', y='2015')
plt.show()
PCA_scatter('file.csv')
我该怎么做?
5条答案
按热度按时间gj3fmq9x1#
seaborn.lmplot
或轴级函数seaborn.regplot
一次性完成整个拟合和绘图图1
图2
图3
tjjdgumg2#
您可以使用
np.polyfit()
和np.poly1d()
。使用相同的x
值估计一次多项式,并添加到由.scatter()
图创建的ax
对象。举个例子:估计一次多项式:
并绘制:
获取:
还提供了直线方程:
agxfikkp3#
另一个选项(使用
np.linalg.lstsq
):pbpqsu0x4#
这涵盖了
plotly
方法其中
fit_line
是pw9qyyiw5#
最好的答案是使用海运。为了补充上面的内容,如果您正在使用循环创建许多图,您仍然可以使用matplotlib
plt.show()将暂停执行,以便您可以一次查看一个图