如何将 3D 散点图与 3D 曲面图结合起来,同时保持曲面图透明,以便我仍然可以看到所有点?
以下代码使用 3D 曲面图绘制 3D 散点图:
import matplotlib.pyplot as plt
import numpy as np
from random import random, seed
from matplotlib import cm
fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(projection='3d') # to work in 3d
x_surf=np.arange(0, 1, 0.01) # generate a mesh
y_surf=np.arange(0, 1, 0.01)
x_surf, y_surf = np.meshgrid(x_surf, y_surf)
z_surf = np.sqrt(x_surf+y_surf) # ex. function, which depends on x and y
ax.plot_surface(x_surf, y_surf, z_surf, cmap=cm.hot, ec='k') # plot a 3d surface plot
n = 100
seed(0) # seed let us to have a reproducible set of random numbers
x=[random() for i in range(n)] # generate n random points
y=[random() for i in range(n)]
z=[random() for i in range(n)]
ax.scatter(x, y, z); # plot a 3d scatter plot
ax.set_xlabel('x label')
ax.set_ylabel('y label')
ax.set_zlabel('z label')
plt.show()
您可以在此处查看一些其他 3D 绘图示例:
http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html
我已将曲面图的颜色从默认值更改为颜色图“热”,以便区分两个图的颜色 - 现在,可以看到曲面图覆盖了散点图,与顺序无关 ...
编辑:要解决该问题,应该在曲面图的颜色图中使用透明度;添加代码: 透明色彩图 并更改线路:
ax.plot_surface(x_surf, y_surf, z_surf, cmap=cm.hot); # plot a 3d surface plot
到
ax.plot_surface(x_surf, y_surf, z_surf, cmap=theCM);
我们得到:
使用siluaty的例子;您可以调整 alpha 值,而不是通过 cmap=theCM 命令使用透明度。这可能会得到你想要的吗?
ax.plot_surface(x_surf, y_surf, z_surf, cmap=cm.hot, alpha=0.2)
正如其他人提到的,如果你想要透明 3D 绘图,只需提供 alpha 参数:
# elve, azim are custom viewpoint/angle
ax = plt.axes(projection='3d', elev=35, azim=-125)
ax.plot_surface(x, y, z, alpha=0.7)
ax.plot(scatter_x, scatter_y, scatter_z, 'b.', markersize=10, label='top')
如果您想在 3D 图的顶部绘制点,请使用 zorder。例如,将生成以下代码:
ax = plt.axes(projection='3d', elev=35, azim=-125)
ax.plot_surface(x, y, z, cmap=plt.cm.coolwarm, linewidth=0.1, zorder=1)
ax.plot(scatter_x, scatter_y, scatter_z, 'b.', markersize=10, label='top', zorder=4)