Matplotlib 中的 3D 图着色：告别单调，拥抱色彩缤纷！

2024-03-06 21:46:09

Matplotlib 中 3D 图的精彩着色

在 Matplotlib 的 3D 绘图世界中，为您的表面增添色彩可以大大增强可视化效果。然而，有时您可能会遇到一个常见的陷阱：单一的颜色，破坏了您的预期彩虹。别担心，这篇文章将引导您逐步解决这个难题，为您提供一个充满色彩的 3D 绘图体验。

问题：单调乏味的 3D 图

使用轮廓来绘制 3D 图时，您可能会发现所有表面都采用相同的颜色。这是因为 Matplotlib 将 z 数据视为一维数组，限制了颜色填充的范围。

解决方案：巧妙地调整 z 数据

为了解决这个问题，我们需要对 z 数据的生成方式进行一些调整。关键步骤如下：

使用 numpy.asarray()： 替换 numpy.array() 使用 numpy.asarray()，它将 z 数据视为二维数组，这是 plot_surface() 函数所期望的。
添加 cmap 参数： 在 plot_surface() 中包含 cmap 参数，它允许您指定颜色图来填充表面。

实践中的代码优化

让我们应用这些更改来完善您的代码：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 调整 z 数据生成
def xyz_ret(file):
    # ...（代码略）
    xyz = np.asarray(xyz) 
    return xyz[:, 0], xyz[:, 1], xyz[:, 2]

# 使用 cmap 参数
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# ...（代码略）
surf = ax.plot_surface(X, Y, Z, rstride=6, cstride=6, cmap=cm.jet,
                        linewidth=0)