Python Seaborn热力图展示相关性矩阵示例
5星
- 浏览量: 0
- 大小:None
- 文件类型：PDF
简介：
本教程详细介绍如何使用Python的Seaborn库创建热力图来直观展示数据的相关性矩阵，帮助读者快速掌握绘制技巧。
在数据分析与机器学习领域内，数据可视化是理解及洞察数据的重要步骤之一。`seaborn` 是一个基于 `matplotlib` 的 Python 数据可视化库，提供了一些高级接口用于创建美观且信息丰富的统计图形，并支持热力图（heatmap）功能。本段落将深入探讨如何利用 `seaborn` 中的 `heatmap` 函数来展示相关性矩阵。
相关性矩阵是一种表示变量间线性关系强度和方向的有效工具，在 `pandas` 库中，我们可以使用 `.corr()` 方法计算数据框内所有列之间的相关系数。这些值范围从 -1 到 1：完全正相关的数值为 1；完全负相关的数值为-1；不相关的则为0。
下面是一个生成随机数并创建相关性矩阵的例子：
```python
import pandas as pd
import numpy as np
# 创建一个包含5列的随机数据框，共有10行。
df = pd.DataFrame(np.random.randn(50).reshape(10, 5))
# 计算各变量之间的相关系数形成矩阵形式
corr = df.corr()
# 使用seaborn库中的heatmap函数进行可视化展示，并设置颜色映射与注释显示数值。
sns.heatmap(corr, cmap=Blues, annot=True)
```
在这个例子中，`cmap=Blues` 参数定义了使用的色彩方案；而 `annot=True` 则使每个单元格内显示出对应的相关系数值。
为了更清晰地展现相关性矩阵的结构，我们可以应用掩码（mask）来隐藏下三角部分。这是因为对角线以下的数据与上半部重复：
```python
# 创建一个大小和相关性矩阵相同的掩码。
mask = np.zeros_like(corr)
# 将掩码的下半部分设置为True以表示不显示这部分数据。
mask[np.tril_indices_from(mask)] = True
# 应用上述创建好的掩码来调整热力图可视化效果
sns.heatmap(corr, cmap=Blues, annot=True, mask=mask.T)
```
此外，我们还可以讨论协方差矩阵。它衡量的是两个变量共同变化的程度：对角线上的元素代表每个变量的变异程度（即方差）；非对角线上的值则表示不同变量间的协同变异性。
```python
import numpy as np
# 定义一个包含四行九列的数据集。
data = [
[11, 22, 33, 44, 55, 66, 77, 88, 99],
[10, 24, 30, 48, 50, 72, 70, 96, 90],
[91, 79, 72, 58, 53, 47, 34, 16, 10],
[55, 20, 98, 19, 17, 10, 77, 89, 14]
]
# 利用numpy的cov()函数计算协方差矩阵。
covariance_matrix = np.cov(data)
# 使用seaborn库中的heatmap功能来展示该数据集：
sns.heatmap(covariance_matrix, center=0, annot=True)
```
在这个例子中，`center=0` 参数确保热图中心值为零；同时还可以设置 `xticklabels=` 和 `yticklabels=` 来指定x轴和y轴的标签。
通过使用 seaborn 的 heatmap 函数来可视化相关性和协方差矩阵，数据科学家能够更深入地理解变量间的关系。这使得数据分析与建模工作更加高效且富有洞察力。
全部评论 (0)
