Matplotlib,避免使用plot_trisurf()中不需要的三角形

时间:2013-04-30 13:12:39

标签: python numpy matplotlib

我有以下代码来创建一个稍后将应用置换字段的圆锥体。在下图所示的图中,您可以看到一些大三角形在顶部绘制,而不是在底部绘制。我相信有一些内部隐藏参数告诉plot_trisurf()三角形应该创建的距离,否则它们也应该在底部创建。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from numpy import linspace, interp, meshgrid, sin, cos, pi
numel_circum = 120
L = 38.
rb, rt = (38., 16.)
t1, t2 = (0, 2*pi)
rav = ( rb - rt )*rt/rb + rt
perav = (t2-t1) * rav
elsize_L = perav / numel_circum
numel_L = int(round(L/elsize_L,0))
ts = linspace(t1, t2, numel_circum)
r = lambda z: interp( z, [0, L], [rb, rt] )
zs = linspace(0, L, numel_L)
ts, zs = meshgrid( ts, zs )
ys = r(zs)*sin(ts)
xs = r(zs)*cos(ts)
# plotting
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_trisurf(xs.flatten(), ys.flatten(), zs.flatten(),
                cmap=cm.jet, linewidth=0.2)
plt.show()

enter image description here

1 个答案:

答案 0 :(得分:4)

你现在可能已经开始或者已经解决了这个问题,但我想我会为未来的游客提出一个答案。

你在较小的圆圈中获得三角形的原因不是最大距离,而是因为这些三角形包含在x,y平面上的点投影的凸包内。

如果您查看plot_trisurf来源(numpy.source(Axes3D.plot_trisurf)),您会看到它每次执行delaunay三角测量,并且没有机会定义三角形或排除不需要的三角形。

两个选项:

正确的方式#1

plot_trisurf的来源复制到您的脚本,然后使用您选择的算法(一些最大边长标准或找到三角形)添加行tri.set_mask(...)(tri是matplotlib.tri.triangulation.Triangulation实例)在三角测量完成后,哪个质心在某个半径范围内......适合您的实际数据)以创建布尔掩码。

正确的方式#2

使用Triangulation(x,y,triangles=...)

定义三角形而不使用delaunay三角测量

快捷方式

将plot_trisurf vmax设置为略低于圆的平面。

*我没有尝试其中任何一个选项