Python绘制等高线_python 等高线

Python绘制等高线

import numpy as np
import matplotlib.pyplot as plt

# 设置图形大小和分辨率
plt.figure

# 生成2D网格坐标
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)

# 在这个网格上生成一个二维随机分布（例如高斯函数）
Z = np.random.normal(loc=0, scale=1, size=X.shape) + np.sin(X**2 + Y**2)

# 绘制等高线密度图
levels = 20  # 设置等高线的层次数量
plt.contourf(X, Y, Z, levels, cmap='RdBu')  # 使用'RdBu'颜色映射方案

# 添加等高线线图
plt.contour(X, Y, Z, levels, colors='k', linewidths=0.5)

# 设置坐标轴标签和标题
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Randomly Generated Contour Plot')

# 添加颜色条以显示不同高度的颜色对应关系
plt.colorbar(label='Z-value')

# 显示图形
plt.show()