python 分段拟合笔记

效果图：

源代码：

import numpy as np
import cv2
import matplotlib.pyplot as plt
from numpy.polynomial.polynomial import Polynomialdef nihe(x_points,y_points,p_id):# 按照 p_id 将 points 分成两组group_0_x = []group_0_y = []group_1_x = []group_1_y = []for i, pid in enumerate(p_id):if pid == 0:group_0_x.append(x_points[i])group_0_y.append(y_points[i])else:group_1_x.append(x_points[i])group_1_y.append(y_points[i])group_0_x = np.array(group_0_x)group_0_y = np