笔者的任务是,[1]根据一堆数据画空间内一个曲面,然后根据[2]另一堆数据画出同一空间内的点,并将点到面的距离标出。面上的点及距离标记作红色,而面下的标记为蓝色。
笔者先作好了曲面,然后循环读取数据,画出点及距离。核心代码如下
# 循环绘制线段,根据点的相对位置确定线段颜色
for i in range(len(points1)):
z1 = points1[i, 2]
z2 = points2[i, 2]
if z1 > z2:
line_color = 'red'
else:
line_color = 'blue'
ax.plot(points1[i, 0], points1[i, 1], z1, marker='o', color=line_color)
ax.plot(points2[i, 0], points2[i, 1], z2, marker='', color=line_color)
ax.plot([points1[i, 0], points2[i, 0]], [points1[i, 1], points2[i, 1]], [z1, z2], color=line_color)
效果图如下:
观察可以感觉出不对劲:明明是在曲面上方的红点,颜色中却混杂着曲面的颜色。
旋转一下得到的图,可以发现红点确实是在曲面上方的。
经过一波资料的查询,发现应该是plot()方法中一个叫"zorder"参数的问题。如果不设置此参数,平面的zorder可能默认比点的zorder要高,导致俯视的时候发现面在点的上方。
基于这一推断,修正后的代码如下(重点设置zorder,将点的zorder与面的zorder显著区分开,使得点的位置被推断在面的上方):
# 循环绘制竖直线段,根据点的相对位置确定线段颜色
for i in range(len(points1)):
z1 = points1[i, 2]
z2 = points2[i, 2]
if z1 > z2:
line_color = 'red'
zorder1=100
else:
line_color = 'blue'
zorder1=1
ax.plot(points1[i, 0], points1[i, 1], z1, marker='o', color=line_color,alpha=1,zorder=zorder1)
ax.plot(points2[i, 0], points2[i, 1], z2, marker='', color=line_color)
ax.plot([points1[i, 0], points2[i, 0]], [points1[i, 1], points2[i, 1]], [z1, z2], color=line_color,zorder=zorder1)
下图是调整后符合预期的作图效果:
可以看到红色的点不会被面所覆盖了。