如何在Python中的函数指定的域中按维度绘制3？
如何在Python中的函数指定的域中按维度绘制3？
提问于 2018-12-17 04:15:26
因此，我尝试用matplotlib.pyplot绘制三维图形，但仅限于边界由函数设置的特定区域。以下是代码
import numpy
from scipy.integrate import quad
import matplotlib.pyplot
import matplotlib.ticker
from mpl_toolkits.mplot3d import Axes3D
def get_x1min(pT, eta, rS):
return (pT * numpy.exp(eta)) / (rS - pT * numpy.exp(- eta))
def get_x2(x1, pT, eta, rS):
return (x1 * pT * numpy.exp(- eta)) / (x1 * rS - pT * numpy.exp(eta))
rS = 1960.0
eta = 0.0
pT = [66.0, 77.5, 89.5, 92.0, 94.0, 96.0, 98.0, 100.0, 103.0, 105.0, 107.0, 109.0, 111.0, 113.0, 118.5, 136.5, 157.5, 182.0, 200.0, 209.5, 241.5, 278.5, 321.0, 370.0, 426.5, 492.0]
fig = matplotlib.pyplot.figure()
ax = Axes3D(fig)
x1_test = numpy.linspace(0.1, 1.0, 20)
x1_test, pT = numpy.meshgrid(x1_test, pT)
x2_test = get_x2(x1_test, pT, eta, rS)
ax.plot_surface(x1_test, pT, x2_test, rstride=1, cstride=1, cmap=matplotlib.pyplot.get_cmap('rainbow'))
matplotlib.pyplot.show()这可以很好地工作，但是域是一个矩形，唯一需要更改的是让x1_test从可以从get_x1min函数获得的最小值开始。有什么方法可以做到吗？
提前感谢！
回答 1
Stack Overflow用户
回答已采纳
发布于 2018-12-17 15:22:56
您可以将不需要的值设置为np.nan，然后不会绘制它们。但请注意，颜色映射似乎存在一些问题。您可以使用vmin和vmax关键字来解决此问题：
mask = x1_test< get_x1min(pT, eta, rS)
x2_test[mask] = numpy.nan
ax.plot_surface(
x1_test, pT, x2_test, rstride=1, cstride=1,
cmap=matplotlib.pyplot.get_cmap('rainbow'),
vmin = numpy.nanmin(x2_test),
vmax = numpy.nanmax(x2_test),
)
matplotlib.pyplot.show()其余的代码将保持不变。结果如下所示：
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/53806093
复制相关文章
相似问题
腾讯云开发者
