时间戳

首页 > 解决方案 > y 需要 polyfit 错误 1d 或 2d 数组

python - y 需要 polyfit 错误 1d 或 2d 数组

问题描述

点击这里查看追溯

我有以下问题：

编写一个 Python 程序来生成使用随机变量（具有高斯分布）和 4 次多项式方程（3x4+x3+3x2+4x+5）之和的数据。使用最小二乘多项式拟合，使用模型弯曲生成的数据，直到您的模型可以准确预测所有值

从以下问题开始：

import random
import numpy as np
import matplotlib.pyplot as plt

def mainFunc():
    poly_coeff=[3,1,3,4,5]
    poly=np.poly1d(poly_coeff)
    print(poly)
    y = poly(random.randint(0,10)) + min(10,max(0,random.gauss(2,3)))
    x=np.arange(-10,10)
    curvefit=np.polyfit(x,y,4)
    y_new=np.polyfit(curvefit,x)
    plt.plot(x,y, '-or')
    plt.plot(x,y_new, '-b')
    plt.show()


mainFunc()

任何人都可以帮助解决正在生成的数组错误吗？

标签： pythonarraysnumpy

解决方案

推荐阅读