matplotlib 等高线热力图
时间: 2023-12-27 08:24:44 浏览: 27
以下是使用Matplotlib创建等高线图和热力图的示例:
1. 创建等高线图:
```python
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.contour(X, Y, Z, levels=20, cmap='jet')
plt.colorbar()
# 添加标题和标签
plt.title('Contour Plot')
plt.xlabel('X')
plt.ylabel('Y')
# 显示图形
plt.show()
```
2. 创建热力图:
```python
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.imshow(Z, extent=[-5, 5, -5, 5], cmap='hot')
plt.colorbar()
# 添加标题和标签
plt.title('Heatmap')
plt.xlabel('X')
plt.ylabel('Y')
# 显示图形
plt.show()
```