登录
首页 >  文章 >  python教程

Python热力图绘制方法大全

时间:2026-03-14 10:20:51 198浏览 收藏

本文系统介绍了Python中绘制热力图的两种主流方法——简洁高效的Seaborn和高度可控的Matplotlib,涵盖从基础语法、参数配置到实际应用场景(如相关性矩阵可视化、密度分布展示)的完整流程,并通过清晰示例代码对比二者在易用性与灵活性上的差异,帮助数据分析师和开发者根据具体需求快速上手、精准呈现二维数据关系。

python绘制热力图有何方法?

Python绘制热力图主要通过可视化库实现,常用方法集中在Matplotlib和Seaborn上,操作简便且效果直观。数据通常以二维数组或DataFrame形式传入,适合展示相关性矩阵、密度分布等场景。

使用Seaborn绘制热力图

Seaborn是基于Matplotlib的高级绘图库,语法简洁,颜色配置美观,适合快速出图。

基本用法如下:

  • 导入seaborn和matplotlib.pyplot
  • 准备数据,如pandas的DataFrame
  • 调用sns.heatmap()函数,设置参数如是否显示数值、颜色映射、是否显示色条等
示例代码:
<code>import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
<h1>示例数据</h1><p>data = pd.DataFrame([[1, 5, 6], [4, 3, 2], [7, 8, 9]], columns=['A', 'B', 'C'])
sns.heatmap(data, annot=True, cmap='coolwarm', center=0)
plt.show()</p></code>

使用Matplotlib绘制基础热力图

Matplotlib提供更底层控制,适合需要自定义布局或集成到复杂图形中的情况。

可通过imshow()函数实现:

  • 传入二维数据矩阵
  • 设置cmap参数选择颜色方案
  • 配合plt.colorbar()显示颜色标尺
示例代码:
<code>import matplotlib.pyplot as plt
import numpy as np
<p>data = np.random.rand(5, 5)
plt.imshow(data, cmap='viridis')
plt.colorbar()
plt.xticks(range(5), ['X1', 'X2', 'X3', 'X4', 'X5'])
plt.yticks(range(5), ['Y1', 'Y2', 'Y3', 'Y4', 'Y5'])
plt.show()</p></code>

结合Pandas进行相关性分析绘图

在数据分析中,常对DataFrame计算corr()得到相关系数矩阵,再用热力图展示。

步骤包括:

  • 使用df.corr()生成相关性矩阵
  • 过滤或处理缺失值
  • 直接将结果传给sns.heatmap()
这样能清晰展现变量之间的正负相关趋势,便于特征选择。

基本上就这些,Seaborn最实用,Matplotlib更灵活,按需选择即可。

终于介绍完啦!小伙伴们,这篇关于《Python热力图绘制方法大全》的介绍应该让你收获多多了吧!欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布文章相关知识,快来关注吧!

资料下载
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>