matplotlib曲面图延伸超过轴限制

Question

如何在Matplotlib中制作一个看起来像

的漂亮抛物面

我能得到的就是这个，

顶部不是＆＃34;切断＆＃34;。我试过将Z数组的所有值都放在顶部的抛物线半径之外，但这会产生非常锯齿状的边缘。有人能帮助我吗？

这是我的代码：

from matplotlib import *
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from pylab import *
import math
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

X = np.arange(-5, 5, 0.1)
Y = np.arange(-5, 5, 0.1)
X, Y = np.meshgrid(X, Y)
Z = (X**2 + Y**2)

ax.set_zlim(-10, 20)

ax.plot_surface(X, Y, Z,  alpha=0.9, rstride=4, cstride=4, linewidth=0.5, cmap=cm.summer)

plt.show()

Answer 1

为了将来参考，我想到了在圆柱坐标中对曲面进行参数化，它看起来就像我想要的那样：

from matplotlib import *
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from pylab import *
import math
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

r = T = np.arange(0, 2*pi, 0.01)
r, T = np.meshgrid(r, T)
#Parametrise it
X = r*np.cos(T)
Y = r*np.sin(T)
Z = r**2

ax.plot_surface(X, Y, Z,  alpha=0.9, rstride=10, cstride=10, linewidth=0.5, cmap=cm.summer)

plt.show()

我想这是有道理的：当使用圆柱形物体时，使用圆柱坐标！

Answer 2

手动数据剪辑

我见过的一种方法是手动剪辑数据;例如您的示例将更新为

from matplotlib import *
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from pylab import *
import math
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

X = np.arange(-5, 5, 0.1)
Y = np.arange(-5, 5, 0.1)
X, Y = np.meshgrid(X, Y)
Z = (X**2 + Y**2)

ax.set_zlim(-10, 20)

for i in range(len(X)):
    for j in range(len(Y)):
        if (Z[j,i] < -10) or (Z[j,i] > 20):
            Z[j,i] = NaN


ax.plot_surface(X, Y, Z,  alpha=0.9, rstride=4, cstride=4, linewidth=0.5, cmap=cm.summer)

plt.show()

注意

对于这种情况，可以使用

简洁地完成

Z[Z>20] = NaN

导致