python曲线拟合代码

            x = np.array(list_x)   # x的点集
            y = np.array(list_y)   # y的点集
            if(len(x)==0):  #当x的点集没有点时,退出
                return 0 
            z1 = np.polyfit(x,y,2)   # 求二次曲线的各个参数
            p1 = np.poly1d(z1)       # 得到二次曲线
            y_pred = p1(x)           # 得到预测值
            for i_ in range(len(x_pred)):
                cv2.circle(img, (int(x_pred[i_]),int(y[i_])), 4, color_list[i], 2) #将拟合得到的点画在原图上