python seaborn heatmap可视化相关性矩阵实例

使用seaborn库的heatmap函数可以方便地将相关性矩阵可视化出来,为数据分析提供很好的帮助。本篇文章将介绍如何使用seaborn库中的heatmap函数实现可视化相关性矩阵。

一、导入所需库

在使用heatmap函数前,需要先导入seaborn库和matplotlib库。

import seaborn as sns

import matplotlib.pyplot as plt

二、生成相关性矩阵

要使用heatmap函数生成相关性矩阵,首先要把数据集生成为相关性矩阵,这可以使用pandas库中的corr函数快速实现。在本例中,我们将使用seaborn库内置的“flights”数据集作为示例数据集。

# 导入数据集

flights = sns.load_dataset("flights")

# 生成相关性矩阵

corr_mat = flights.corr()

# 查看相关性矩阵

print(corr_mat)

输出相关性矩阵如下所示:

| | year | passengers |

|--------|-------|-------------|

| year | 1.000000 | 0.960195 |

| passengers | 0.960195 | 1.000000 |

三、可视化相关性矩阵

生成相关性矩阵后,即可使用seaborn库中的heatmap函数进行可视化。heatmap函数提供了许多参数,可以根据需求进行调整。在本例中,我们将使用默认参数进行可视化。

# 可视化相关性矩阵

sns.heatmap(corr_mat)

# 显示可视化图形

plt.show()

上述代码将相关性矩阵可视化为一个矩阵形式,不同颜色表示不同的相关程度,如下图所示:

![heatmap图片展示](https://ai-studio-static-online.cdn.bcebos.com/6a953018e5e74a7fb7c221f903ed105fb26c447865164e66b90e4b69e7c90b77)

四、调整可视化效果

heatmap函数提供了许多参数,可以调整可视化效果。例如,可以设置热图的颜色映射以及行列标签的大小等。

1.调整颜色映射

在使用heatmap函数可视化相关性矩阵时,有时需要调整颜色映射以便更好地突出相关性程度。可以通过设置cmap参数来进行调整。

# 调整颜色映射为蓝色调

sns.heatmap(corr_mat, cmap="Blues")

# 显示可视化图形

plt.show()

上述代码将颜色映射设置为蓝色调,并且生成的热图如下所示:

![调整颜色映射之后的heatmap图片展示](https://ai-studio-static-online.cdn.bcebos.com/b11a7c019f874ebf9cac5cf882285a43c9c91d583d8a4335a54b259218b41fef)

2.调整行列标签大小

当数据集比较大时,可视化出来的热图行列标签可能会相对较小,需要进行调整。可以通过设置xticklabels、yticklabels等参数来进行调整。

# 调整行列标签大小

sns.heatmap(corr_mat, cmap="Blues", xticklabels=corr_mat.columns.values, yticklabels=corr_mat.columns.values, annot=True, annot_kws={"size": 14})

# 显示可视化图形

plt.show()

上述代码将行列标签大小设置为14,并且加入数值注释,生成的热图如下所示:

![调整行列标签大小之后的heatmap图片展示](https://ai-studio-static-online.cdn.bcebos.com/aa96d8e9a4c54a5b93dbb0c09f27815b30b638c0158c43df8b5ce364b06a0e14)

结论

本文介绍了如何使用seaborn库中的heatmap函数可视化相关性矩阵。通过调整heatmap函数的参数,可以生成不同样式的热图。在进行数据分析时,相关性矩阵可视化是一个非常重要的工具。感谢您的阅读!

后端开发标签