温馨提示:本文翻译自stackoverflow.com,查看原文请点击:numpy - How can I interpolate missing values (undefined areas) of a 3D surface plot (2D array) using Python?
numpy python scipy interpolation surface

numpy - 如何使用Python插入3D表面图(2D数组)的缺失值(未定义区域)?

发布于 2020-03-27 12:05:08

在使用Numpy和matplotlib的Python 3.7中,我想为以下方程式绘制3D曲面:

f(x,y)= sin(x)sin(y)/(x * y)

显然,该函数在x = 0y = 0时未定义

为了计算和绘制此图,我有以下代码,当前正在Jupyter Notebook中运行:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
%matplotlib notebook

f = lambda x, y: np.sin(x)*np.sin(y)/(x*y)

xs, ys = np.mgrid[-np.pi:np.pi:31j, -np.pi:np.pi:31j]
zs = f(xs, ys)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.plot_surface(X=xs, Y=ys, Z=zs)

请注意该图,它缺少值: 在此处输入图片说明

如何对缺失值进行插值,以使图形显得平滑?

查看更多

查看更多

提问者
clockelliptic
被浏览
71
clockelliptic 2019-07-05 00:45

Scipy具有可以执行此操作的插值模块。依靠以上内容(在问题发布中),此代码可以在下一个单元格中运行:

from scipy import interpolate

# integer arrays for indexing
x_indx, y_indx = np.meshgrid(np.arange(0, zs.shape[1]),
                             np.arange(0, zs.shape[0]))

# mask all invalid values
zs_masked = np.ma.masked_invalid(zs)

# retrieve the valid, non-Nan, defined values
valid_xs = x_indx[~zs_masked.mask]
valid_ys = y_indx[~zs_masked.mask]
valid_zs = zs_masked[~zs_masked.mask]

# generate interpolated array of z-values
zs_interp = interpolate.griddata((valid_xs, valid_ys), valid_zs.ravel(),
                                 (x_indx, y_indx), method='cubic')

# finally, plot the data 
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.plot_surface(X=xs, Y=ys, Z=zs_interp)

返回下图: 在此处输入图片说明

请注意,此代码已针对可读性和可理解性(而非内存效率)进行了优化。重新优化此代码以提高内存效率是一项琐碎的任务,留给读者