Seaborn库学习之heatmap()函数

Seaborn库学习之heatmap(函数)

一、简介

seaborn.heatmap是Seaborn库中用于绘制热图(Heatmap)的函数。热图是一种数据可视化技术,通过颜色的变化来展示数据矩阵中的数值大小。这种图表非常适合展示数值数据的分布和关系,尤其是在数据科学和统计分析中。

二、语法和参数

seaborn.heatmap的基本语法如下:

python 复制代码
seaborn.heatmap(data, annot=True, fmt=".2f", cmap="YlGnBu", cbar=True, cbar_kws=None, xticklabels=True, yticklabels=True, mask=None, square=False, linewidths=0, linecolor="white", ax=None, cbar_ax=None, cbar_pad=None, **kwargs)

主要参数说明:

  • data: 要展示的数据,通常是一个二维数组或DataFrame。⭐⭐⭐
  • annot: 是否在热图中显示数据值,默认为True。⭐⭐⭐
  • fmt: 数据值的格式化字符串,默认为".2f"。
  • cmap: 颜色映射,用于定义热图中的颜色渐变,默认为"YlGnBu"。⭐⭐⭐
  • cbar: 是否显示颜色条,默认为True。⭐⭐⭐
  • xticklabels: 是否显示x轴标签,默认为True。⭐⭐⭐
  • yticklabels: 是否显示y轴标签,默认为True。⭐⭐⭐
  • mask: 用于隐藏数据的布尔数组或数组。⭐⭐⭐
  • square: 是否将每个单元格设置为正方形,默认为False。
  • linewidths: 单元格边框的宽度。
  • linecolor: 单元格边框的颜色。
  • ax: 指定绘制热图的坐标轴。
  • cbar_ax: 指定颜色条的坐标轴。
  • cbar_pad: 颜色条和热图之间的间距。

三、实例

3.1 基本热图
python 复制代码
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np

# 创建一个数据矩阵
data = np.random.rand(10, 10)

# 绘制热图
sns.heatmap(data, annot=True, fmt=".2f", cmap="YlGnBu")

plt.show()

输出:

3.2 带颜色条的热图
python 复制代码
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np

# 创建一个数据矩阵
data = np.random.rand(10, 10)

# 绘制带颜色条的热图
sns.heatmap(data, annot=True, fmt=".2f", cmap="YlGnBu", cbar=True)

plt.show()

输出:

3.3 隐藏部分数据
python 复制代码
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np

# 创建一个数据矩阵
data = np.random.rand(10, 10)

# 定义一个掩码数组
mask = np.zeros((10, 10), dtype=bool)
mask[1:5, 1:5] = True

# 绘制热图并隐藏部分数据
sns.heatmap(data, annot=True, fmt=".2f", cmap="YlGnBu", mask=mask)

plt.show()

输出:

四、注意事项

  1. 数据类型data参数需要是一个二维数组或DataFrame。
  2. 颜色映射 :可以通过cmap参数自定义热图中的颜色渐变效果。
  3. 格式化字符串fmt参数可以控制数据值的显示格式。
  4. 掩码 :使用mask参数可以隐藏或显示特定的数据区域。
  5. 坐标轴标签 :可以通过xticklabelsyticklabels参数控制是否显示坐标轴标签。
相关推荐
鸡鸭扣34 分钟前
Docker:3、在VSCode上安装并运行python程序或JavaScript程序
运维·vscode·python·docker·容器·js
paterWang1 小时前
基于 Python 和 OpenCV 的酒店客房入侵检测系统设计与实现
开发语言·python·opencv
东方佑1 小时前
使用Python和OpenCV实现图像像素压缩与解压
开发语言·python·opencv
饮长安千年月2 小时前
Linksys WRT54G路由器溢出漏洞分析–运行环境修复
网络·物联网·学习·安全·机器学习
红花与香菇2____2 小时前
【学习笔记】Cadence电子设计全流程(二)原理图库的创建与设计(上)
笔记·嵌入式硬件·学习·pcb设计·cadence·pcb工艺
神秘_博士2 小时前
自制AirTag,支持安卓/鸿蒙/PC/Home Assistant,无需拥有iPhone
arm开发·python·物联网·flutter·docker·gitee
Moutai码农3 小时前
机器学习-生命周期
人工智能·python·机器学习·数据挖掘
一天八小时4 小时前
Docker学习进阶
学习·docker·容器
前端没钱4 小时前
前端需要学习 Docker 吗?
前端·学习·docker
小白教程4 小时前
python学习笔记,python处理 Excel、Word、PPT 以及邮件自动化办公
python·python学习·python安装