Matplotlib库学习之plot_surface函数

Matplotlib库学习之plot_surface函数

一、简介

plot_surface 是 Matplotlib 库中的一个函数，用于在三维空间中绘制表面图。它能够根据给定的 x, y 坐标网格和相应的 z 值，生成一个三维表面图，非常适合用来展示变量间的交互作用和三维数据的分布情况。

二、语法和参数

plot_surface 函数的基本语法如下：

ax.plot_surface(X, Y, Z, *args, **kwargs)

• X, Y, Z：分别代表 x, y, z 坐标的二维数组，其中 X 和 Y 通常由 np.meshgrid 生成。
• *args：额外的参数，用于传递给绘制函数。
• **kwargs：额外的关键字参数，可以用于设置颜色、透明度等属性。

常见的关键字参数包括：

• cmap：用于指定颜色映射表。
• edgecolor：用于指定表面边缘的颜色。
• lw 或 linewidth：用于设置表面线条的宽度。
• alpha：用于设置表面的透明度。

三、实例

3.1 绘制基本的表面图

import matplotlib.pyplot as plt
import numpy as np

# 创建新的图和坐标轴
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 生成网格数据
X = np.linspace(-5, 5, 100)
Y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))

# 绘制表面图
surf = ax.plot_surface(X, Y, Z, cmap='viridis')

# 添加颜色条
fig.colorbar(surf)

# 显示图表
plt.show()

输出：一个使用 viridis颜色映射表的三维表面图，展示了 $\sin (\mathrm{sqrt}({\mathrm{x}}^2+{\mathrm{y}}^2))$ 的形状。

3.2 自定义表面图的颜色和透明度

# 继续使用上面的 X, Y, Z 数据

# 绘制自定义颜色和透明度的表面图
surf = ax.plot_surface(X, Y, Z, cmap='plasma', alpha=0.6, linewidth=0)

# 显示图表
plt.show()

输出：一个使用 ‘plasma’ 颜色映射表和 0.6 透明度的三维表面图。

四、注意事项

• 在使用 plot_surface 之前，需要确保已经导入了 Matplotlib 的 pyplot 模块，并创建了三维坐标轴。
• X, Y, Z 数组的维度必须相同，且它们应该形成网格数据。
• cmap 参数可以接受 Matplotlib 提供的任何预定义颜色映射表，也可以自定义颜色映射。
• alpha 参数的值范围是 0 到 1，0 表示完全透明，1 表示完全不透明。
• 可以通过 fig.colorbar(surface_object) 添加颜色条，以便于观察数据的高低变化。
• 绘制三维图形时，性能可能会受到数据量大小的影响。对于非常大的数据集，可能需要优化或使用更高效的可视化方法。

五、附录

5.1 matplotlib内部提供的颜色映射表

• 顺序（Sequential）色彩映射表

  顺序色彩映射表用于展示单一方向上的数据变化，通常从低到高，颜色由浅入深。

  ‘Blues’, ‘BrBG’, ‘BuGn’, ‘BuPu’, ‘CMRmap’, ‘GnBu’, ‘Greens’, ‘Greys’, ‘OrRd’, ‘Oranges’, ‘PRGn’, ‘PiYG’, ‘PuBu’, ‘PuBuGn’, ‘PuOr’, ‘PuRd’, ‘Purples’, ‘RdBu’, ‘RdGy’, ‘RdPu’, ‘RdYlBu’, ‘RdYlGn’, ‘Reds’, ‘Spectral’, ‘Wistia’, ‘YlGn’, ‘YlGnBu’, ‘YlOrBr’, ‘YlOrRd’

• 顺序（二）（Sequential (2)）色彩映射表

  顺序（二）色彩映射表提供了另一种顺序变化的色阶，适用于不同的数据展示需求。

  ‘afmhot’, ‘autumn’, ‘binary’, ‘bone’, ‘brg’, ‘bwr’, ‘cool’, ‘coolwarm’, ‘copper’, ‘cubehelix’, ‘gist_earth’, ‘gist_gray’, ‘gist_heat’, ‘gist_ncar’, ‘gist_rainbow’, ‘gist_stern’, ‘gist_yarg’

• 定性（Qualitative）色彩映射表

  定性色彩映射表用于区分不同的类别，颜色之间没有顺序或数值关系。

  ‘Accent’, ‘Dark2’, ‘Paired’, ‘Pastel1’, ‘Pastel2’, ‘Set1’, ‘Set2’, ‘Set3’, ‘tab10’, ‘tab20’, ‘tab20b’, ‘tab20c’

• 杂项（Miscellaneous）色彩映射表

  杂项色彩映射表包括一些不适用于其他分类的特殊颜色映射，用于特定的可视化需求。

  ‘flag’, ‘gnuplot’, ‘gnuplot2’, ‘gray’, ‘hot’, ‘hsv’, ‘jet’, ‘nipy_spectral’, ‘ocean’, ‘pink’, ‘prism’, ‘rainbow’, ‘seismic’, ‘spring’, ‘summer’, ‘terrain’, ‘winter’

• 感知均匀顺序（Perceptually Uniform Sequential）色彩映射表

  感知均匀顺序色彩映射表提供了在视觉上均匀分布的渐变，适合表示顺序数据，且对色盲用户友好。

  ‘magma’, ‘inferno’, ‘plasma’, ‘viridis’, ‘cividis’, ‘twilight’, ‘twilight_shifted’, ‘turbo’

• 反转颜色映射表（Reversed Colormaps）

  反转颜色映射表是现有色彩映射表的反转版本，提供了从深到浅的颜色变化。

  ‘magma_r’, ‘inferno_r’, ‘plasma_r’, ‘viridis_r’, ‘cividis_r’, ‘twilight_r’, ‘twilight_shifted_r’, ‘turbo_r’, ‘Blues_r’, ‘BrBG_r’, ‘BuGn_r’, ‘BuPu_r’, ‘CMRmap_r’, ‘GnBu_r’, ‘Greens_r’, ‘Greys_r’, ‘OrRd_r’, ‘Oranges_r’, ‘PRGn_r’, ‘PiYG_r’, ‘PuBu_r’, ‘PuBuGn_r’, ‘PuOr_r’, ‘PuRd_r’, ‘Purples_r’, ‘RdBu_r’, ‘RdGy_r’, ‘RdPu_r’, ‘RdYlBu_r’, ‘RdYlGn_r’, ‘Reds_r’, ‘Spectral_r’, ‘Wistia_r’, ‘YlGn_r’, ‘YlGnBu_r’, ‘YlOrBr_r’, ‘YlOrRd_r’, ‘afmhot_r’, ‘autumn_r’, ‘binary_r’, ‘bone_r’, ‘brg_r’, ‘bwr_r’, ‘cool_r’, ‘coolwarm_r’, ‘copper_r’, ‘cubehelix_r’, ‘flag_r’, ‘gist_earth_r’, ‘gist_gray_r’, ‘gist_heat_r’, ‘gist_ncar_r’, ‘gist_rainbow_r’, ‘gist_stern_r’, ‘gist_yarg_r’, ‘gnuplot_r’, ‘gnuplot2_r’, ‘gray_r’, ‘hot_r’, ‘hsv_r’, ‘jet_r’, ‘nipy_spectral_r’, ‘ocean_r’, ‘pink_r’, ‘prism_r’, ‘rainbow_r’, ‘seismic_r’, ‘spring_r’, ‘summer_r’, ‘terrain_r’, ‘winter_r’, ‘Accent_r’, ‘Dark2_r’, ‘Paired_r’, ‘Pastel1_r’, ‘Pastel2_r’, ‘Set1_r’, ‘Set2_r’, ‘Set3_r’, ‘tab10_r’, ‘tab20_r’, ‘tab20b_r’, ‘tab20c_r’

要获取 Matplotlib 中所有注册的颜色映射表的列表，可以使用以下代码：

from matplotlib import colormaps
list(colormaps)

此外，还可以通过 matplotlib.cm.get_cmap 函数获取命名的颜色映射表对象，并使用它来访问颜色值 。还可以创建自定义颜色映射表，通过 ListedColormap 或 LinearSegmentedColormap 类来实现颜色映射的创建和操作 。