scipy 光滑散点图

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

def SmoothData(x, y):
    x_smooth = np.linspace(x.min(), x.max(), 3000)
    y_smooth = make_interp_spline(x, y)(x_smooth)
    return x_smooth, y_smooth

fig = plt.figure(figsize=(12, 5))
ax1, ax2 = fig.subplots(2, 1)

x = np.arange(1, 101, 20)
y = x ** 3
x_smooth, y_smooth = SmoothData(x, y)

ax2.plot(x_smooth, y_smooth)
ax2.scatter(x, y, s=12)

ax1.plot(x, y, '-o', lw=2)

ax1.grid(alpha=0.3)
ax2.grid(alpha=0.3)
plt.show()

1.jpg