### 如何在 Python 中设置等高线图的颜色 在 Python 的 Matplotlib 库中，可以通过 `ax.contour()` 或 `plt.contour()` 函数来绘制等高线图，并通过参数控制颜色。以下是具体方法： #### 使用单一颜色 如果希望所有的等高线具有相同的颜色，可以在 `ax.contour()` 或 `plt.contour()` 函数中添加 `colors` 参数并指定颜色名称或十六进制颜色码。例如： ```python import numpy as np import matplotlib.pyplot as plt def fun(x, y): return (1 - x / 2 + x ** 2 + y ** 3) * np.exp(-x ** 2 - y ** 2) n = 256 x = np.linspace(-3, 3, n) y = np.linspace(-3, 3, n) X, Y = np.meshgrid(x, y) CS = plt.contour(X, Y, fun(X, Y), colors='blue') # 设置所有线条为蓝色 plt.clabel(CS, inline=True, fontsize=8) # 添加标注 plt.title('Contour Plot with Single Color') plt.xlabel('X-axis') plt.ylabel('Y-axis') plt.show() ``` 上述代码设置了所有等高线的颜色为蓝色[^1]。 --- #### 使用多种颜色 如果不满足于单色等高线，可以使用默认配色方案或者自定义颜色映射（colormap）。对于填充的等高线图，推荐使用 `plt.contourf()` 方法；而对于非填充的等高线图，也可以通过 colormap 来实现多色效果。 ##### 非填充等高线图 通过 `cmap` 参数指定颜色映射表： ```python CS = plt.contour(X, Y, fun(X, Y), cmap='viridis') # 使用 'viridis' 色彩映射 plt.colorbar(CS) # 添加颜色条 plt.title('Contour Plot with Colormap') plt.xlabel('X-axis') plt.ylabel('Y-axis') plt.show() ``` 此处使用了 `'viridis'` 这一预定义的颜色映射表[^3]。 ##### 填充等高线图 同样支持通过 `cmap` 参数调整颜色： ```python plt.contourf(X, Y, fun(X, Y), cmap='plasma') # 使用 'plasma' 色彩映射 plt.colorbar() # 添加颜色条 plt.title('Filled Contour Plot with Colormap') plt.xlabel('X-axis') plt.ylabel('Y-axis') plt.show() ``` 这里展示了如何应用 `'plasma'` 色彩映射到填充等高线图上。 --- #### 平滑处理与插值 当原始数据点较少时，为了获得更平滑的等高线图，通常需要对数据进行插值处理。这一步骤可通过 SciPy 提供的插值工具完成。例如： ```python from scipy.interpolate import griddata # 构造稀疏的数据点 points = np.random.rand(100, 2) * 6 - 3 # 随机生成100个点 values = fun(points[:, 0], points[:, 1]) # 插值得到密集网格上的高度值 xi, yi = np.meshgrid(np.linspace(-3, 3, 100), np.linspace(-3, 3, 100)) zi = griddata(points, values, (xi, yi), method='cubic') # 绘制平滑后的等高线图 plt.contour(xi, yi, zi, levels=15, linewidths=0.5, colors='k') plt.contourf(xi, yi, zi, levels=15, cmap="RdBu_r") # 使用红蓝渐变色彩映射 plt.colorbar() plt.title('Smoothed Contour Plot with Interpolation') plt.xlabel('X-axis') plt.ylabel('Y-axis') plt.show() ``` 此部分说明了如何利用插值技术提升图像质量[^2]。 --- ### 总结 - 单一颜色：通过 `colors` 参数设定。 - 多种颜色：借助 `cmap` 参数选择合适的色彩映射表。 - 数据不足时需考虑插值以提高平滑度。