matplotlib保存热图

在使用matplotlib绘制热图之后，可以使用savefig方法保存热图。

具体步骤如下：

1. 导入matplotlib库

   import matplotlib.pyplot as plt

2. 使用imshow函数绘制热图，设置参数

   plt.imshow(data, cmap='hot')

3. 添加颜色条

   plt.colorbar()

4. 保存热图

   plt.savefig('heatmap.png')

完整代码示例：

import matplotlib.pyplot as plt
import numpy as np

# 创建随机数据
data = np.random.rand(10, 10)

# 绘制热图
plt.imshow(data, cmap='hot')

# 添加颜色条
plt.colorbar()

# 保存热图
plt.savefig('heatmap.png')

保存后可在当前路径下看到一张名为heatmap.png的热图。