matplotlib 绘制热力图
时间: 2023-10-10 17:05:04 浏览: 168
要使用matplotlib绘制热力图,可以使用Python中的pyplot模块。在这个例子中,需要导入matplotlib.pyplot和numpy模块。
下面是一个绘制热力图的示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建一个二维数组作为数据
harvest = np.array([[0.8, 2.4, 2.5, 3.9, 0.0, 4.0, 0.0],
[2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0],
[1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0],
[0.6, 0.0, 0.3, 0.0, 3.1, 0.0, 0.0],
[0.7, 1.7, 0.6, 2.6, 2.2, 6.2, 0.0],
[1.3, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1],
[0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3]])
# 使用imshow函数绘制热力图
plt.imshow(harvest)
# 调整图像布局
plt.tight_layout()
# 显示图像
plt.show()
```
这个示例代码中,我们首先创建了一个二维数组harvest作为数据,然后使用imshow函数绘制热力图。最后,通过调用tight_layout函数调整图像布局,并使用show函数显示图像。
这样就可以使用matplotlib绘制热力图了。
阅读全文