使用Axes3D.plot_wireframe和Axes3D.scatter进行绘图:线框隐藏了散点

时间:2019-04-30 12:17:35

标签: python matplotlib

我使用Axes3d.plot_wireframe绘制磁盘的离散表面,并在同一图形上使用Axes3d.scatter叠加一个与磁盘中心相对应的点(附图)。我的问题是,只有在配置文件中看到光盘时,该点才可见。我希望无论视图如何都可以看到它。我还加入了我的脚本的一部分(对不起,如果没有按原样出现)。

我尝试更改脚本中图表的顺序,尝试在两个图表中添加一个'zorder'自变量,还尝试更改散点图中的标记大小,但这没有用。

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D
import pylab as plt
import numpy as np

# data to build the disc
r = np.linspace(0, 5e-3, 30)
psi = np.linspace(0, 2*np.pi, 30)
n = np.array([0.70710678, 0, 0.70710678])
e2 = np.array([0.70710678, 0, -0.70710678])
e3 = np.array([0, 1, 0])
Oc = np.array([0, 0, 0])
R, PSI = np.meshgrid(r, psi)
OpP = np.zeros((len(r),len(psi),3), dtype=float)
OpP[:,:,0] = np.zeros((len(r),len(psi)), dtype=float)
OpP[:,:,1] = R*np.cos(PSI)  
OpP[:,:,2] = R*np.sin(PSI)  
Mrot = np.zeros((3,3), dtype=float)
Mrot[0,:] = n
Mrot[1,:] = e2
Mrot[2,:] = e3
Ocx = Oc[0]; Ocy = Oc[1]; Ocz = Oc[2]
xp = Ocx + np.dot(OpP, Mrot)[:,:,0]
yp = Ocy + np.dot(OpP, Mrot)[:,:,1]
zp = Ocz + np.dot(OpP, Mrot)[:,:,2]

# Plotting
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(zp,yp,xp, label='piston')
ax.scatter(Ocz, Ocy, Ocx, zdir='z', s=40, color='red', label='Oc')
plt.gca().invert_yaxis()
plt.legend()
plt.show()

光盘的配置文件视图: enter image description here

定向光盘: enter image description here

1 个答案:

答案 0 :(得分:0)

可以设置绘图参数的某些值以获得更好的绘图。尝试用以下代码替换相关的代码行:

ax.plot_wireframe(zp,yp,xp, label='piston', \
            linewidth=0.5, \
            alpha=0.7)
ax.scatter(Ocz, Ocy, Ocx, zdir='z', \
            s=80, \
            color='red', label='Oc')

3d image