使用scipy将样条线拟合到具有重复x的数据(Python)

import numpy as np
from scipy.interpolate import UnivariateSpline
import matplotlib.pyplot as plt

# 创建具有重复x值的数据
x = np.array([1, 2, 2, 3, 4, 4, 4, 5])
y = np.array([1, 2, 3, 3, 2, 1, 0, 0])

# 创建样条插值
spline = UnivariateSpline(x, y, k=3)

# 创建用于绘图的x值
xnew = np.linspace(x.min(), x.max(), 1000)

# 计算插值的y值
ynew = spline(xnew)

# 绘制原始数据和插值
plt.plot(x, y, 'o', label='Original data')
plt.plot(xnew, ynew, label='Interpolated data')
plt.legend()
plt.show()