python 等高线热力图
时间: 2023-12-25 09:30:11 浏览: 155
Python绘制热力图示例
4星 · 用户满意度95%
以下是使用Matplotlib库在Python中绘制等高线热力图的示例:
```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.contourf(X, Y, Z, cmap='hot')
plt.colorbar()
# 添加等高线线条
plt.contour(X, Y, Z, colors='black')
# 添加标题和标签
plt.title('Contour Heatmap')
plt.xlabel('X')
plt.ylabel('Y')
# 显示图形
plt.show()
```
这段代码首先创建了一个二维网格,然后计算了每个点的函数值。接下来,使用`contourf`函数绘制了等高线热力图,并使用`colorbar`函数添加了颜色条。然后,使用`contour`函数添加了等高线线条。最后,添加了标题和标签,并使用`show`函数显示图形。
阅读全文