【Python】matplotlib(散点图)添加趋势线

【参考】

1. 如何在python matplotlib点(散点图)中添加趋势线？

2. np.polyfit()与np.poly1d()将点拟合成曲线

 

【补充】

import matplotlib as mpl
import matplotlib.pyplot as plt

def plot_trendline(x, y, n):
    mpl.pylab.plot(x, y,  'ko')
    parameter = np.polyfit(x, y, n) # n=1为一次函数，返回函数参数
    f = np.poly1d(parameter) # 拼接方程
    mpl.pylab.plot(x, f(x),"r--")
    plt.show()