Matplotlib 等高线图是用于在二维平面上可视化三维数据的强大工具。本文将深入探讨 matplotlib 等高线图的世界，探索其各种特性、定制选项和实际应用。无论您是数据科学家、研究人员还是可视化爱好者，本指南将帮助您掌握使用 matplotlib 创建惊人等高线图的艺术。
Matplotlib 等高线推荐文章
理解 Matplotlib 等高线图
Matplotlib 等高线图用于在二维平面上表示三维数据。它们由连接相等值点的等高线组成，类似于地形图。这些图特别适合可视化标量场，例如温度分布、压力梯度或高程图。
让我们从创建一个基本的 matplotlib 等高线图的例子开始:
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建轮廓图
plt.figure(figsize=(10, 8))
contour = plt.contour(X, Y, Z)
plt.colorbar(contour)
plt.title("基础 Matplotlib 轮廓图 - how2matplotlib.com")
plt.xlabel("X 轴")
plt.ylabel("Y 轴")
plt.show()
输出:
在这个例子中，我们使用 numpy 生成一个示例数据集，并使用 matplotlib 创建一个基本的轮廓图。plt.contour() 函数是创建轮廓图的关键，我们将在本文中探讨它的各种参数和选项。
自定义轮廓级别
matplotlib 等高线图最重要的方面之一是能够自定义等高线级别。默认情况下，matplotlib 会自动选择等高线级别的数量和数值，但您可以手动指定它们，以便更好地控制您的可视化效果。
以下是如何设置自定义轮廓级别的示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建带有自定义级别的轮廓图
plt.figure(figsize=(10, 8))
levels = [-0.75, -0.5, -0.25, 0, 0.25, 0.5, 0.75]
contour = plt.contour(X, Y, Z, levels=levels)
plt.colorbar(contour)
plt.title("Matplotlib 轮廓图与自定义级别 - how2matplotlib.com")
plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.show()
输出:
在这个例子中，我们使用 plt.contour() 函数中的 levels 参数指定自定义轮廓级别。这使您可以专注于特定的数值范围或创建更具视觉吸引力的图形。
为轮廓线添加标签
为了使您的 matplotlib 等高线图更具信息性，您可以为等高线添加标签。这在您想要显示每条等高线所代表的确切值时特别有用。
这是一个如何为轮廓线添加标签的示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建带标签的轮廓图
plt.figure(figsize=(10, 8))
contour = plt.contour(X, Y, Z)
plt.clabel(contour, inline=True, fontsize=8)
plt.colorbar(contour)
plt.title("Matplotlib 带标签的轮廓图 - how2matplotlib.com")
plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.show()
输出:
在这个例子中，我们使用 plt.clabel() 函数为轮廓线添加标签。inline=True 参数确保标签直接放置在轮廓线上，而 fontsize 控制标签文本的大小。
创建填充轮廓图
常规等高线图使用线条表示相等的值，而填充等高线图使用彩色区域表示值的范围。这对于可视化连续数据分布特别有效。
以下是如何创建填充等高线图的示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建填充等高线图
plt.figure(figsize=(10, 8))
contourf = plt.contourf(X, Y, Z, cmap='viridis')
plt.colorbar(contourf)
plt.title("Matplotlib 填充等高线图 - how2matplotlib.com")
plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.show()
输出:
在这个例子中，我们使用 plt.contourf() 函数来创建一个填充等高线图。cmap 参数允许您为填充区域指定一个颜色映射。
结合轮廓图和填充轮廓图
为了更全面的可视化，您可以将常规等高线与填充等高区域结合起来。这种方法提供了等高线的精确值信息和填充等高线的整体分布视图。
这是一个如何结合轮廓图和填充轮廓图的示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建组合等高线图
plt.figure(figsize=(10, 8))
contourf = plt.contourf(X, Y, Z, cmap='viridis', alpha=0.7)
contour = plt.contour(X, Y, Z, colors='black', linewidths=0.5)
plt.clabel(contour, inline=True, fontsize=8)
plt.colorbar(contourf)
plt.title("组合 Matplotlib 等高线图 - how2matplotlib.com")
plt.xlabel("X 轴")
plt.ylabel("Y 轴")
plt.show()
输出:
在这个例子中，我们首先使用 plt.contourf() 创建一个填充等高线图，然后使用 plt.contour() 叠加常规等高线。plt.contourf() 中的 alpha 参数控制填充区域的透明度，使等高线可见。
自定义轮廓颜色
Matplotlib 提供了广泛的选项来定制等高线图的颜色。您可以使用预定义的颜色映射，创建自定义颜色方案，甚至为每个等高线级别指定单独的颜色。
这是一个使用自定义颜色图进行轮廓图的示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
从 matplotlib.colors 导入 LinearSegmentedColormap
# 生成示例数据
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))
# 创建一个自定义颜色映射
colors = ['蓝色', '青色', '黄色', '红色']
n_bins = 100
cmap = LinearSegmentedColormap.from_list('custom_cmap', colors, N=n_bins)
# 创建带有自定义颜色映射的轮廓图
plt.figure(figsize=(10, 8))
contour = plt.contour(X, Y, Z, cmap=cmap)
plt.colorbar(contour)
plt.title("Matplotlib 轮廓图与自定义颜色映射 - how2matplotlib.com")
plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.show()
输出:
在这个例子中，我们使用 LinearSegmentedColormap.from_list() 创建一个自定义的颜色映射，并通过 cmap 参数将其应用于我们的轮廓图。
添加颜色条
色条是 matplotlib 等高线图的重要组成部分，因为它提供了一个将颜色映射到数值的图例。虽然我们在之前的示例中包含了色条，但让我们探索一些额外的自定义选项。
这是自定义颜色条的一个示例：
导入 numpy 作为 np
导入 matplotlib.pyplot 作为 plt
# 生成示例数据
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))
# 创建带自定义颜色条的轮廓图
plt.figure(figsize=(12, 8))
contour = plt.contour(X, Y, Z, cmap='coolwarm')
cbar = plt.colorbar(contour, orientation='horizontal', pad=0.1, aspect=30)
cbar.set_label('值', fontsize=12)
cbar.ax.tick_params(labelsize=10)
plt.title("Matplotlib 轮廓图与自定义颜色条 - how2matplotlib.com")
plt.xlabel("X轴")
plt.ylabel("Y轴")
plt.show()
输出:
在这个例子中，我们通过改变颜色条的方向、调整其位置和大小，以及修改标签和刻度属性来自定义颜色条。