jupyter绘制热力图并保存
时间: 2023-10-17 22:10:15 浏览: 100
Python数据相关系数矩阵和热力图轻松实现教程
5星 · 资源好评率100%
要在 Jupyter Notebook 中绘制热力图并保存,可以使用 Python 的 matplotlib 库和 seaborn 库。
下面是一个简单的示例代码,可以绘制一个随机生成的 10x10 的矩阵的热力图:
```python
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
# 生成 10x10 的随机矩阵
data = np.random.rand(10, 10)
# 绘制热力图
sns.heatmap(data, cmap='YlGnBu')
# 保存图片
plt.savefig('heatmap.png')
```
在上面的示例代码中,我们首先使用 numpy 库生成一个 10x10 的随机矩阵,然后使用 seaborn 库绘制热力图,并将 colormap 设置为 "YlGnBu"。最后,使用 matplotlib 库保存生成的图片。
需要注意的是,如果要在 Jupyter Notebook 中显示图片,还需要使用 matplotlib 库的 `plt.show()` 函数。完整代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
# 生成 10x10 的随机矩阵
data = np.random.rand(10, 10)
# 绘制热力图
sns.heatmap(data, cmap='YlGnBu')
# 显示图片
plt.show()
# 保存图片
plt.savefig('heatmap.png')
```
阅读全文