Python进行二维插值的代码

最新推荐文章于 2024-11-11 10:00:00 发布

MATLAB代码顾问

最新推荐文章于 2024-11-11 10:00:00 发布

阅读量1k

点赞数 12

CC 4.0 BY-SA版权

文章标签： python 开发语言

本文链接：https://blog.csdn.net/corn1949/article/details/136293334

本文详细介绍了如何在Python中利用scipy.interpolate库的griddata函数进行二维插值，通过生成随机数据点、定义插值网格并执行cubic方法，最后展示了插值后的图像和散点图。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在Python中，我们可以使用scipy.interpolate库中的griddata函数进行二维插值。

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

# 创建原始数据点
x = np.random.uniform(-2, 2, 100)
y = np.random.uniform(-2, 2, 100)
z = x * np.exp(-x ** 2 - y ** 2)

# 定义插值的网格点
xi = np.linspace(-2.0, 2.0, 100)
yi = np.linspace(-2.0, 2.0, 100)

# 创建网格
xi, yi = np.meshgrid(xi, yi)

# 进行二维插值
zi = griddata((x, y), z, (xi, yi), method='cubic')

# 绘制插值后的图像
plt.imshow(zi, vmin=zi.min(), vmax=zi.max(), origin='lower',
           extent=[-2, 2, -2, 2], aspect='auto')
plt.scatter(x, y, c=z, edgecolor='k')
plt.colorbar()
plt.xlabel('X')
plt.ylabel('Y')
plt.title('2D Interpolation')
plt.show()