python画出相关性热力图并保存高清图片
时间: 2023-09-09 12:10:15 浏览: 44
要画出相关性热力图并保存高清图片,我们需要使用 Python 中的 Seaborn 库和 Matplotlib 库。以下是实现的步骤:
1. 导入所需库:
```python
import seaborn as sns
import matplotlib.pyplot as plt
```
2. 创建相关性矩阵:
```python
corr_matrix = df.corr()
```
其中,`df` 为包含数据的 Pandas DataFrame 对象。
3. 使用 Seaborn 库中的 `heatmap()` 函数绘制热力图:
```python
sns.heatmap(corr_matrix, annot=True)
```
其中,`annot=True` 表示在热力图上显示相关系数的值。
4. 设置图像属性:
```python
plt.title('Correlation Heatmap')
plt.tight_layout()
```
可以设置热力图的标题,并使用 `tight_layout()` 函数使图像更紧凑。
5. 保存图像:
```python
plt.savefig('correlation_heatmap.png', dpi=300)
```
使用 `savefig()` 函数将图像保存为指定格式的文件。其中,`dpi=300` 表示设置图像的分辨率为 300 dpi,以获得更高质量的图像。
完整代码如下:
```python
import seaborn as sns
import matplotlib.pyplot as plt
# 创建相关性矩阵
corr_matrix = df.corr()
# 绘制热力图
sns.heatmap(corr_matrix, annot=True)
# 设置图像属性
plt.title('Correlation Heatmap')
plt.tight_layout()
# 保存图像
plt.savefig('correlation_heatmap.png', dpi=300)
```
以上就是如何使用 Python 画出相关性热力图并保存高清图片的方法。