自留-Python:线性拟合(直线+曲线)

这篇具有很好参考价值的文章主要介绍了自留-Python:线性拟合(直线+曲线)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

使用最小二乘法的线性拟合,自留代码

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import rcParams
import math
import matplotlib

读取数据

#(1)读取excel数据
df = pd.read_excel(r'C:\Users\Administrator\Desktop\worklist.xlsx')
data = np.array(df)

#(2)自己设定组数
X=np.array([0,1.34,2.25,4.67,7.2,9.6,12.79,15.61]) 
Y=np.array([0,12.5,25,50,100,200,400,800])

直线拟合

#定义直线拟合函数
def linear_regression(x, y): 
    N = len(x)
    sumx = sum(x)
    sumy = sum(y)
    sumx2 = sum(x**2)
    sumxy = sum(x*y)
 
    A = np.mat([[N, sumx], [sumx, sumx2]])
    b = np.array([sumy, sumxy])
 
    return np.linalg.solve(A, b)
 
a10, a11 = linear_regression(X, Y)


# 生成拟合直线的绘制点
_X1 = np.arange (0,20,0.01)
_Y1 = np.array([a10 + a11 * x for x in _X1])

#画图
plt.figure(figsize=(6,6))
plt.plot(_X1, _Y1, 'b', linewidth=2) 
plt.legend(bbox_to_anchor=(1,0),loc="lower left")
plt.title("y = {} + {}x".format(a10, a11)) #标题
plt.show()

python 直线拟合,大数据,python

 曲线拟合

# 生成系数矩阵A
def gen_coefficient_matrix(X, Y): 
    N = len(X)
    m = 3
    A = []
    # 计算每一个方程的系数
    for i in range(m):
        a = []
        # 计算当前方程中的每一个系数
        for j in range(m):
            a.append(sum(X ** (i+j)))
    A.append(a)
    return A
 
# 计算方程组的右端向量b
def gen_right_vector(X, Y): 
    N = len(X)
    m = 3
    b = []
    for i in range(m):
        b.append(sum(X**i * Y))
    return b
 
A = gen_coefficient_matrix(X, Y) 
b = gen_right_vector(X, Y)
 
a0, a1, a2 = np.linalg.solve(A, b)
print(a0,a1,a2)

#绘制拟合曲线
_X = np.arange(0, 20, 0.01) 
_Y = np.array([a0 + a1*x + a2*x**2 for x in _X])


#画图
plt.figure(figsize=(10,6))
plt.plot(X,Y,'o',markersize=10,label='Hou等(2017)')

#plt.plot(_X, _Y, 'b', linewidth=2,label="$I$ = {:.2f} + {:.2e}$n$ +{:.2e}$n^2$ ".format(a0, a1, a2))    #{:.2f}等用于保留小数

plt.plot(_X, _Y, 'b', linewidth=2,label="式(1) ") 
#plt.gca().invert_yaxis()
plt.legend(fontsize=16,frameon=False)
#plt.legend(bbox_to_anchor=(0.5,-0.35),loc=10,ncol=2,frameon=False) #图框
#plt.title("AI = {} + {}n + {}$n^2$ ".format(a0, a1, a2))   #标题
plt.show()

python 直线拟合,大数据,python

 文章来源地址https://www.toymoban.com/news/detail-521238.html

到了这里,关于自留-Python:线性拟合(直线+曲线)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包