将散点图与曲面图相结合

问题描述 投票:0回答:3

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

python matplotlib scatter geometry-surface
3个回答
20
投票

以下代码使用 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);

我们得到:


18
投票

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

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

0
投票

正如其他人提到的,如果你想要透明 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)

© www.soinside.com 2019 - 2024. All rights reserved.