• 大小: 734B
    文件类型: .py
    金币: 1
    下载: 0 次
    发布日期: 2021-05-10
  • 语言: Python
  • 标签: Python  拟合直线  

资源简介

Python最小二乘法拟合直线,采用了2种不同的方式,一种是直接计算,另一种是调用numpy.linalg.solve()

资源截图

代码片段和文件信息

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import numpy as np
import math

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)

def linear_regression2(x y):
    N = len(x)
    A = s

评论

共有 条评论