python散点图拟合曲线如何求拟合_python散点图最佳拟合直线的代码

你可以用numpy的polyfit。我使用以下（您可以安全地删除关于确定系数和误差界限的位，我只是认为它看起来不错）：#!/usr/bin/python3

import numpy as np

import matplotlib.pyplot as plt

import csv

with open("example.csv", "r") as f:

data = [row for row in csv.reader(f)]

xd = [float(row[0]) for row in data]

yd = [float(row[1]) for row in data]

# sort the data

reorder = sorted(range(len(xd)), key = lambda ii: xd[ii])

xd = [xd[ii] for ii in reorder]

yd = [yd[ii] for ii in reorder]

# make the scatter plot

plt.scatter(xd, yd, s=30, alpha=0.15, marker='o')

# determine best fit line

par = np.polyfit(xd, yd, 1, full=True)

slope=par[0][0]

intercept=par[0][1]

xl = [min(xd), max(xd)]

yl = [slope*xx + intercept for xx in xl]

# coefficient of determination, plot text

v