二维直方图的样条插值- Python

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

# 生成随机数据
np.random.seed(0)
x = np.random.rand(100)
y = np.random.rand(100)

# 创建二维直方图
hist, xedges, yedges = np.histogram2d(x, y, bins=10, range=[[0, 1], [0, 1]])

# 计算网格点
x_grid = np.linspace(0, 1, 10)
y_grid = np.linspace(0, 1, 10)
X_grid, Y_grid = np.meshgrid(x_grid, y_grid)

# 使用样条插值
f = interp2d(xedges[:-1], yedges[:-1], hist.T, kind='cubic')
Z_interp = f(x_grid, y_grid)

# 绘制结果
plt.figure(figsize=(10, 5))
plt.subplot(1, 2, 1)
plt.imshow(hist.T, extent=[0, 1, 0, 1], origin='lower', cmap='viridis')
plt.title('Original Histogram')

plt.subplot(1, 2, 2)
plt.imshow(Z_interp, extent=[0, 1, 0, 1], origin='lower', cmap='viridis')
plt.title('Spline Interpolated Histogram')

plt.show()