最小二乘拟合二维直线

目录

  • [1. 原理概述](#1. 原理概述)
  • [2. python实现](#2. python实现)
  • [3. matlab实现](#3. matlab实现)
  • [4. C++实现](#4. C++实现)

爬虫网站自重。

1. 原理概述

平面直线的表达式为:
y = k x + b (1) y=kx+b \tag{1} y=kx+b(1)

假设有 n n n个点 ( x i , y i ) ( 0 ≤ i < n ) (x_i, y_i)(0≤i<n) (xi,yi)(0≤i<n),于是目标函数为:
f ( x ) = ∑ i = 1 n k x i + b − y i (2) f(x)=\sum_{i=1}^n kx_i+b-y_i\tag{2} f(x)=i=1∑nkxi+b−yi(2)

分别求 f ( x ) f(x) f(x)对于 k 、 b k、b k、b的偏导数:
δ f δ k = ∑ i = 1 n 2 ( k x i + b − y i ) x i = 2 ∑ i = 1 n k x i 2 + b x i − x i y i = 2 ( k ∑ i = 1 n x i 2 + b ∑ i = 1 n x i − ∑ i = 1 n x i y i ) (3) \frac{\delta f}{\delta k}=\sum_{i=1}^n 2(kx_i+b-y_i)x_i=2\sum_{i=1}^n kx_i^2+bx_i-x_iy_i \\=2(k \sum_{i=1}^n x_i^2+b\sum_{i=1}^n x_i- \sum_{i=1}^n x_iy_i)\tag{3} δkδf=i=1∑n2(kxi+b−yi)xi=2i=1∑nkxi2+bxi−xiyi=2(ki=1∑nxi2+bi=1∑nxi−i=1∑nxiyi)(3)

δ f δ b = ∑ i = 1 n 2 ( k x i + b − y i ) = 2 ∑ i = 1 n k x i + b − y i = 2 ( k ∑ i = 1 n x i + b n − ∑ i = 1 n y i ) (4) \frac{\delta f}{\delta b}=\sum_{i=1}^n 2(kx_i+b-y_i)=2\sum_{i=1}^n kx_i+b-y_i \\=2(k \sum_{i=1}^n x_i+bn- \sum_{i=1}^n y_i)\tag{4} δbδf=i=1∑n2(kxi+b−yi)=2i=1∑nkxi+b−yi=2(ki=1∑nxi+bn−i=1∑nyi)(4)

令(3)式和(4)式偏导数为0,得到一个二元一次方程组:

{ k ∑ i = 1 n x i 2 + b ∑ i = 1 n x i = ∑ i = 1 n x i y i k ∑ i = 1 n x i + b n = ∑ i = 1 n y i (5) \begin{cases} k \sum_{i=1}^n x_i^2+b\sum_{i=1}^n x_i= \sum_{i=1}^n x_iy_i\\ k \sum_{i=1}^n x_i+bn= \sum_{i=1}^n y_i \end{cases} \tag{5} {k∑i=1nxi2+b∑i=1nxi=∑i=1nxiyik∑i=1nxi+bn=∑i=1nyi(5)


A = ∑ i = 1 n x i 2 , B = ∑ i = 1 n x i , C = ∑ i = 1 n x i y i , D = ∑ i = 1 n y i A=\sum_{i=1}^n x_i^2,B=\sum_{i=1}^n x_i,C=\sum_{i=1}^n x_iy_i,D=\sum_{i=1}^n y_i A=i=1∑nxi2,B=i=1∑nxi,C=i=1∑nxiyi,D=i=1∑nyi

于是有:

{ A k + B b = C B k + b n = D (6) \begin{cases} Ak+Bb=C\\ Bk+bn=D \end{cases} \tag{6} {Ak+Bb=CBk+bn=D(6)

解以上方程组得到 k 、 b k、b k、b
{ k = n C − B D n A − B 2 b = D A − B C n A − B 2 (7) \begin{cases} k=\frac{nC-BD}{nA-B^2}\\ b=\frac{DA-BC}{nA-B^2} \end{cases} \tag{7} {k=nA−B2nC−BDb=nA−B2DA−BC(7)

2. python实现

python 复制代码
# -----------------------------------拟合直线y=kx+b--------------------------------
n = points.shape[0]
A = sum(x * x)
B = sum(x)
C = sum(x * y)
D = sum(y)
k = (n * C - B * D) / (n * A - B * B)
b = (D * A - B * C) / (n * A - B * B)

3. matlab实现

cpp 复制代码
n = pc.Count;
A = sum(x.^2);
B = sum(x);
C = sum(x.*y);
D = sum(y);
k = (n * C - B * D) / (n * A - B * B);
b = (D * A - B * C) / (n * A - B * B);

4. C++实现

cpp 复制代码
    int n = cloud->points.size();
	double A = 0.0, B = 0.0, C = 0.0, D = 0.0;

	for (int i = 0; i < n; i++)
	{
		A += pow(cloud->points[i].x, 2);
		B += cloud->points[i].x;
		C += cloud->points[i].x * cloud->points[i].y;
		D += cloud->points[i].y;
		
	}

	double k = (n * C - B * D) / (n * A - B * B);
	double b = (D * A - B * C) / (n * A - B * B);
相关推荐
前端拾光者几秒前
利用D3.js实现数据可视化的简单示例
开发语言·javascript·信息可视化
The_Ticker2 分钟前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
程序猿阿伟2 分钟前
《C++ 实现区块链:区块时间戳的存储与验证机制解析》
开发语言·c++·区块链
有Li17 分钟前
跨视角差异-依赖网络用于体积医学图像分割|文献速递-生成式模型与transformer在医学影像中的应用
人工智能·计算机视觉
傻啦嘿哟20 分钟前
如何使用 Python 开发一个简单的文本数据转换为 Excel 工具
开发语言·python·excel
大数据编程之光24 分钟前
Flink Standalone集群模式安装部署全攻略
java·大数据·开发语言·面试·flink
初九之潜龙勿用24 分钟前
C#校验画布签名图片是否为空白
开发语言·ui·c#·.net
爪哇学长38 分钟前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
Dola_Pan42 分钟前
C语言:数组转换指针的时机
c语言·开发语言·算法
ExiFengs42 分钟前
实际项目Java1.8流处理, Optional常见用法
java·开发语言·spring