Matplotlib绘制等高线图：全面指南与实例|极客教程

Matplotlib绘制等高线图：全面指南与实例

参考：Contour Plot using Matplotlib – Python

Matplotlib是Python中最流行的数据可视化库之一，它提供了强大的工具来创建各种类型的图表，包括等高线图。等高线图是一种二维图形，用于表示三维数据，特别适合展示地形、温度分布、压力场等连续变化的数据。本文将深入探讨如何使用Matplotlib绘制等高线图，从基础概念到高级技巧，为您提供全面的指导。

1. 等高线图基础

等高线图（Contour Plot）是一种用于表示三维数据的二维图形。它通过在平面上绘制等值线来表示数据的第三个维度。每条等值线代表具有相同值的所有点的集合。

1.1 创建基本等高线图

让我们从一个简单的例子开始，创建一个基本的等高线图：

import numpy as np
import matplotlib.pyplot as 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.title('Basic Contour Plot - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

在这个例子中，我们首先创建了一个网格数据，然后使用plt.contour()函数绘制等高线。colorbar用于显示颜色刻度。

1.2 填充等高线

除了线条，我们还可以填充等高线之间的区域：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.cos(X) * np.sin(Y)

plt.figure(figsize=(10, 8))
contourf = plt.contourf(X, Y, Z, cmap='viridis')
plt.title('Filled Contour Plot - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contourf)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这里我们使用plt.contourf()函数来创建填充的等高线图。cmap参数用于指定颜色映射。

2. 自定义等高线

2.1 设置等高线级别

我们可以通过指定级别来控制等高线的数量和位置：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = X*np.exp(-X**2 - Y**2)

plt.figure(figsize=(10, 8))
levels = [-0.5, -0.4, -0.3, -0.2, -0.1, 0, 0.1, 0.2, 0.3, 0.4, 0.5]
contour = plt.contour(X, Y, Z, levels=levels)
plt.clabel(contour, inline=True, fontsize=8)
plt.title('Contour Plot with Custom Levels - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

在这个例子中，我们使用levels参数指定了等高线的具体值。plt.clabel()函数用于在等高线上添加标签。

2.2 自定义等高线颜色

我们可以为每个等高线指定不同的颜色：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

plt.figure(figsize=(10, 8))
levels = [0.5, 1, 2, 3, 4, 5]
colors = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd', '#8c564b']
contour = plt.contour(X, Y, Z, levels=levels, colors=colors)
plt.clabel(contour, inline=True, fontsize=8)
plt.title('Contour Plot with Custom Colors - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这里我们使用colors参数为每个等高线指定了不同的颜色。

3. 高级等高线技巧

3.1 绘制不规则数据的等高线

有时我们需要处理不规则分布的数据点：

import numpy as np
import matplotlib.pyplot as plt
from scipy.interpolate import griddata

# 生成随机数据点
np.random.seed(0)
x = np.random.rand(100) * 4 - 2
y = np.random.rand(100) * 4 - 2
z = x*np.exp(-x**2 - y**2)

# 创建网格
xi = np.linspace(-2, 2, 100)
yi = np.linspace(-2, 2, 100)
Xi, Yi = np.meshgrid(xi, yi)

# 插值
Zi = griddata((x, y), z, (Xi, Yi), method='cubic')

plt.figure(figsize=(10, 8))
contour = plt.contour(Xi, Yi, Zi, levels=15)
plt.scatter(x, y, c='r', s=5)
plt.clabel(contour, inline=True, fontsize=8)
plt.title('Contour Plot of Irregular Data - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何使用scipy.interpolate.griddata函数对不规则分布的数据点进行插值，然后绘制等高线图。

3.2 3D等高线图

我们还可以在3D空间中绘制等高线图：

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

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))

fig = plt.figure(figsize=(12, 8))
ax = fig.add_subplot(111, projection='3d')
contour = ax.contour(X, Y, Z, levels=20, cmap='viridis')
ax.set_title('3D Contour Plot - how2matplotlib.com')
ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
ax.set_zlabel('Z-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何在3D空间中绘制等高线图，使用mpl_toolkits.mplot3d模块。

4. 等高线图的应用

4.1 地形图

等高线图常用于表示地形：

import numpy as np
import matplotlib.pyplot as plt

def terrain(x, y):
    return np.sin(x*0.1) * np.cos(y*0.1) + np.random.rand(x.shape[0], y.shape[1])*0.1

x = np.linspace(0, 50, 100)
y = np.linspace(0, 50, 100)
X, Y = np.meshgrid(x, y)
Z = terrain(X, Y)

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, Z, levels=20, cmap='terrain')
plt.title('Terrain Contour Map - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子模拟了一个简单的地形图，使用terrain颜色映射来增强视觉效果。

4.2 温度分布图

等高线图也可以用来表示温度分布：

import numpy as np
import matplotlib.pyplot as plt

def temperature(x, y):
    return 20 + 5*np.sin(x*0.1) + 3*np.cos(y*0.1) + np.random.rand(x.shape[0], y.shape[1])

x = np.linspace(0, 50, 100)
y = np.linspace(0, 50, 100)
X, Y = np.meshgrid(x, y)
Z = temperature(X, Y)

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, Z, levels=20, cmap='coolwarm')
plt.title('Temperature Distribution - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour, label='Temperature (°C)')
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子模拟了一个温度分布图，使用coolwarm颜色映射来区分冷热区域。

5. 等高线图的美化

5.1 添加标签和注释

为等高线图添加标签和注释可以提高其可读性：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

plt.figure(figsize=(12, 8))
contour = plt.contour(X, Y, Z, levels=10, cmap='RdYlBu')
plt.clabel(contour, inline=True, fontsize=8, fmt='%1.2f')
plt.title('Contour Plot with Labels - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)

# 添加注释
plt.annotate('Maximum', xy=(0, 1.5), xytext=(1, 2),
             arrowprops=dict(facecolor='black', shrink=0.05))

plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何使用plt.clabel()添加等高线标签，以及如何使用plt.annotate()添加注释。

5.2 自定义颜色映射

创建自定义颜色映射可以让你的等高线图更具特色：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 - Y**2

# 创建自定义颜色映射
colors = ['darkblue', 'blue', 'lightblue', 'white', 'yellow', 'orange', 'red']
n_bins = 100
cmap = LinearSegmentedColormap.from_list('custom_cmap', colors, N=n_bins)

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, Z, levels=20, cmap=cmap)
plt.title('Contour Plot with Custom Colormap - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何创建和使用自定义颜色映射，使用LinearSegmentedColormap.from_list()函数。

6. 等高线图与其他图表的结合

6.1 等高线图与散点图结合

将等高线图与散点图结合可以同时展示连续数据和离散数据：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

# 生成随机点
np.random.seed(0)
points_x = np.random.uniform(-3, 3, 50)
points_y = np.random.uniform(-3, 3, 50)
points_z = np.sin(points_x) * np.cos(points_y)

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, Z, levels=20, cmap='viridis', alpha=0.7)
scatter = plt.scatter(points_x, points_y, c=points_z, cmap='viridis', edgecolors='w')
plt.title('Contour Plot with Scatter Points - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何将等高线图与散点图结合，使用相同的颜色映射来保持一致性。

6.2 等高线图与向量场结合

等高线图还可以与向量场结合，用于表示如风向、流体流动等：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 20)
y = np.linspace(-3, 3, 20)
X, Y = np.meshgrid(x, y)
Z = X*np.exp(-X**2 - Y**2)

# 计算梯度
dx, dy = np.gradient(Z)

plt.figure(figsize=(12, 8))
contour = plt.contour(X, Y, Z, levels=15, cmap='RdYlBu')
quiver = plt.quiver(X, Y, dx, dy, scale=50)
plt.title('Contour Plot with Vector Field - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何将等高线图与向量场结合。我们使用np.gradient()计算梯度，然后用plt.quiver()绘制向量场。

7. 等高线图的性能优化

在处理大型数据集时，等高线图的绘制可能会变得很慢。以下是一些优化技巧：

7.1 使用适当的分辨率

选择合适的网格分辨率可以在保持图像质量和提高性能之间取得平衡：

import numpy as np
import matplotlib.pyplot as plt
import time

def plot_contour(resolution):
    x = np.linspace(-5, 5, resolution)
    y = np.linspace(-5, 5, resolution)
    X, Y = np.meshgrid(x, y)
    Z = np.sin(np.sqrt(X**2 + Y**2))

    start_time = time.time()
    plt.figure(figsize=(10, 8))
    contour = plt.contour(X, Y, Z, levels=20)
    plt.title(f'Contour Plot (Resolution: {resolution}x{resolution}) - how2matplotlib.com')
    plt.xlabel('X-axis')
    plt.ylabel('Y-axis')
    plt.colorbar(contour)
    end_time = time.time()

    print(f"Resolution {resolution}x{resolution}: {end_time - start_time:.4f} seconds")

plot_contour(100)
plot_contour(500)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子比较了不同分辨率下绘制等高线图的时间。通常，较低的分辨率可以显著提高性能，同时仍能保持足够的细节。

7.2 使用简化的等高线

对于大型数据集，可以考虑减少等高线的数量：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-5, 5, 500)
y = np.linspace(-5, 5, 500)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y) * np.exp(-0.1*(X**2 + Y**2))

plt.figure(figsize=(12, 8))
contour = plt.contour(X, Y, Z, levels=10)  # 使用较少的等高线
plt.title('Simplified Contour Plot - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子使用了较少的等高线级别，可以加快绘图速度，特别是对于大型数据集。

8. 等高线图的交互性

Matplotlib提供了一些方法来增加等高线图的交互性：

8.1 使用鼠标悬停显示数值

我们可以使用mpld3库来创建交互式的等高线图：

import numpy as np
import matplotlib.pyplot as plt
import mpld3

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

fig, ax = plt.subplots(figsize=(10, 8))
contour = ax.contourf(X, Y, Z, levels=20, cmap='viridis')
ax.set_title('Interactive Contour Plot - how2matplotlib.com')
ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
plt.colorbar(contour)

mpld3.enable_notebook()
mpld3.display(fig)

这个例子创建了一个交互式的等高线图，你可以使用鼠标悬停在图上查看具体的数值。注意，这需要在Jupyter Notebook或类似环境中运行。

8.2 动态更新等高线图

我们可以创建一个动态更新的等高线图：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

fig, ax = plt.subplots(figsize=(10, 8))

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)

def update(frame):
    Z = np.sin(X + frame/10) * np.cos(Y)
    ax.clear()
    contour = ax.contourf(X, Y, Z, levels=20, cmap='viridis')
    ax.set_title(f'Dynamic Contour Plot (Frame {frame}) - how2matplotlib.com')
    ax.set_xlabel('X-axis')
    ax.set_ylabel('Y-axis')
    return contour,

ani = FuncAnimation(fig, update, frames=range(100), interval=50, blit=True)
plt.colorbar()
plt.show()

这个例子创建了一个动态更新的等高线图，展示了如何使用FuncAnimation来创建动画效果。

9. 等高线图的常见问题及解决方案

9.1 处理缺失数据

有时数据集中可能存在缺失值，我们需要适当处理：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 20)
y = np.linspace(-3, 3, 20)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

# 添加一些缺失值
Z[5:10, 5:10] = np.nan

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, Z, levels=20, cmap='viridis', extend='both')
plt.title('Contour Plot with Missing Data - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()

Output:

Matplotlib绘制等高线图：全面指南与实例

这个例子展示了如何处理包含缺失值（NaN）的数据。extend='both'参数确保颜色映射覆盖所有数据，包括超出范围的值。

9.2 处理不规则边界

对于具有不规则边界的数据，我们可以使用掩码：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)

# 创建一个圆形掩码
mask = X**2 + Y**2 <= 4

plt.figure(figsize=(12, 8))
contour = plt.contourf(X, Y, np.ma.masked_where(~mask, Z), levels=20, cmap='viridis')
plt.title('Contour Plot with Irregular Boundary - how2matplotlib.com')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.colorbar(contour)
plt.show()