将散点图与曲面图结合起来

时间:2013-03-05 17:09:01

标签: python matplotlib scatter geometry-surface

如何将3D散点图与3D曲面图结合起来,同时保持曲面图透明,以便我仍能看到所有点?

2 个答案:

答案 0 :(得分:17)

要在同一图表中组合各种类型的图表,您应该使用函数

plt.hold(真)。

以下代码绘制了具有3D曲面图的3D散点图:

from mpl_toolkits.mplot3d import *
import matplotlib.pyplot as plt
import numpy as np
from random import random, seed
from matplotlib import cm


fig = plt.figure()
ax = fig.gca(projection='3d')               # to work in 3d
plt.hold(True)

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);    # 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()

结果:

http://s9.postimage.org/ge0wb8kof/3d_scatter_surface_plt.gif

你可以在这里看到一些带有3d图的其他例子:
http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html

为了区分两个图的颜色,我已经将曲面图的颜色从默认颜色更改为“热图” - 现在,可以看到曲面图独立地覆盖散点图订单 ...

编辑:要解决此问题,应在曲面图的色彩图中使用透明度;添加代码: Transparent colormap 并改变线:

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);

我们得到:

http://s16.postimage.org/5qiqn0p5h/3d_scatter_surface_plt_transparent.gif

答案 1 :(得分:13)

使用siluaty的例子;而不是通过cmap = theCM命令使用透明度,您可以调整alpha值。这可能会得到你想要的东西吗?

ax.plot_surface(x_surf, y_surf, z_surf, cmap=cm.hot, alpha=0.2)