最小二乘法多项式拟合代码

好的，以下是最小二乘法多项式拟合的 Python 代码：

import numpy as np
def polyfit(x, y, degree):
    results = {}
    coeffs = np.polyfit(x, y, degree)
     # Polynomial Coefficients
    results['polynomial'] = coeffs.tolist()
    
    # r-squared
    p = np.poly1d(coeffs)
    yhat = p(x)                      
    ybar = np.sum(y)/len(y)          
    ssreg = np.sum((yhat-ybar)**2)   
    sstot = np.sum((y - ybar)**2)    
    results['r_squared'] = ssreg / sstot
    
    return results

其中，x 和 y 分别为输入的数据点的横纵坐标，degree 为拟合多项式的次数。函数返回一个字典，包含拟合多项式的系数和 R-squared 值。