一文弄懂Seaborn绘制热力图

2024-04-17 22:12

本文主要是介绍一文弄懂Seaborn绘制热力图,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

1. 引言

在本文中,我们将使用Seaborn库来以heatmap热力图的形式来表示数据。我们将重点介绍如何创建它,以及如何更改其颜色,调整对应字体大小等等。

闲话少说,我们直接开始吧!

2. 什么是热力图?

Heatmap,中文称为热力图,是一种数据可视化技术,它通过颜色的变化来表示数据的大小或密度。在热力图中,数据大小的变化通常会被表示为颜色渐变,从而在二维空间上形成一种“热度”的分布图。

颜色的深浅通常用来表示数据的量值大小,颜色越深表示数值越高,反之则越低。这种图表非常适合展示地理信息、用户行为、科学测量等数据的分布情况。

3. 安装Seaborn库

要在Python中创建heatmap,可以使用Seaborn库进行操作,该库建立在Matplotlib库的基础之上。Seaborn库提供了一个高级的数据可视化抽象接口方便我们可以绘制各种矩阵数据。

想要在代码中使用该库,一般通过pip命令进行该库的安装,如下:

pip install seaborn

运行结果如下:
在这里插入图片描述

4. 基本用法

好的,现在让我们创建一个热力图,首先让我们导入我们所需的库,如下:

import numpy as np
import seaborn as sb
import matplotlib.pyplot as plt

我们导入了numpy模块,用以生成给定范围内的随机数数组,进而将该数组绘制为热力图。

data = np.random.rand(4, 6)

上述代码将创建一个2维数组,行数为4,列数为6;上述数组中的值表示用来画热力图的数值,接着我们将使用Seaborn库中的heatmap函数来绘制热力图,相应的绘制代码如下:

heat_map = sb.heatmap(data)

接着让我们使用matplotlib库中的plt.show()函数来展示上述热力图,如下:


plt.show()

运行后结果如下:
在这里插入图片描述

5.删除刻度标签

在数据可视化中,tick labels刻度标签是指在图表中x轴、y轴上显示的标签值,用于指示轴上每个刻度的具体数值或类别。Seaborn库在绘制热力图的时候默认会添加相应的刻度标签, 如果我们想要删除上述刻度标签我们可以通过设置xticklabel或者yticklabel的属性值来进行控制。代码如下:

heat_map = sb.heatmap(data, xticklabels=False, yticklabels=False)
plt.show()

运行后结果如下:
在这里插入图片描述

6. 设置x轴文本标签

我们可以使用Matplotlib库中的pyplot.label函数来为上述热力图添加x轴上的文本标签说明,代码如下:

heat_map = sb.heatmap(data)
plt.xlabel("Values on X axis")
plt.show()

运行后结果如下:
在这里插入图片描述

7. 删除y轴刻度标签

一般来说,Seaborn库也默认在y轴上添加相应的刻度标签,我们可以通过设置yticklabels=False,来实现保留x轴上的默认刻度标签,仅删除y轴上的刻度标签,代码如下:

heat_map = sb.heatmap(data, yticklabels=False)
plt.show()

运行后结果如下:

在这里插入图片描述

8.设置y轴文本标签

参考前面章节,我们也可以类似地使用Matplotlib库中的pyplot.ylabel来实现绘制热力图上y轴的文本标签,代码如下:

data = np.random.rand(4, 6)
heat_map = sb.heatmap(data)
plt.ylabel('Values on Y axis')
plt.show()

运行后结果如下:
在这里插入图片描述

9. 设置热力图颜色

我们也可以通过使用heatmap中的cmap属性来通过改变调色板color map进而实现更改seaborn中绘制heatmap的颜色。代码如下:

data = np.random.rand(4, 6)
heat_map = sb.heatmap(data,cmap="YlGnBu")
plt.show()

运行后结果如下:
在这里插入图片描述

10. 设置热力图网格显示数值

如果我们需要在heatmap上显示具体的文本数值,我们可以使用heatmap函数中的annot参数控制。如果设置annot=True, 则热力图上每个单元格会展示对应的文本数值。代码如下:

data = np.random.rand(4, 6)
heat_map = sb.heatmap(data, annot=True)
plt.show()

运行结果如下:
在这里插入图片描述

11. 调整显示字体大小

我们可以使用seabornfont_scale属性来调整heatmap显示文本中的字体大小,如下所示:

data = np.random.rand(4, 6)
sb.set(font_scale=2)
heat_map = sb.heatmap(data, annot=True)
plt.show()

增大显示文本字体大小后,热力图的效果如下所示:
在这里插入图片描述

12.控制热力图调色板

热力图中的调色板如下图中圆圈所示:
在这里插入图片描述
事实上,heatmap函数中的参数cbar是一个Bool型变量,主要用来控制是否应该显示上述调色板。默认来说,上述变量设置为True,表示默认在画热力图的时候会展示相应的调色板。我们也可以使用以下代码,来删除右侧图中的调色板,代码如下:

data = np.random.rand(4, 6)
sb.set(font_scale=2)
heat_map = sb.heatmap(data, annot=True,cbar=False)
plt.show()

运行后结果如下:
在这里插入图片描述

13. 调色板添加文本标签

为了给热力图中的调色板添加相应的文本标签,用来说明文字提示。我们一般使用变量参数cbar_kws来控制,代码如下:

data = np.random.rand(4, 6)
heat_map = sb.heatmap(data, annot=True, cbar_kws={'label': 'My Colorbar'})
plt.show()

运行后结果如下:
在这里插入图片描述

14. 调整调色板显示位置

参考上一章节,参数cbar_kws除了控制为调色板添加自定义文本外,还可以控制调色板水平和垂直显示。虽然默认情况下,调色板为垂直显示,我们可以通过以下代码来控制水平显示调色板,代码如下:

data = np.random.rand(4, 6)
heat_map = sb.heatmap(data, annot=True, cbar_kws={'label': 'My Colorbar', 'orientation': 'horizontal'})
plt.show()

运行后结果如下:
在这里插入图片描述

15. 自定义网格线

自定义Seaborn热力图中的网格线可以增强数据可视化的可读性和美观性。调整线宽和线条颜色可以明显地分离数据点,使热力图更加直观和直观。我们可以调整分隔单元格的线条宽度,并改变它们的颜色,以使单元格之间的区别更加清晰。

data = np.random.rand(4, 6)
sns.heatmap(df, annot=True, cmap='coolwarm', linewidths=0.5, linecolor='black')
plt.show()

运行后结果如下:
在这里插入图片描述
上图中的热力图现在有更明显的网格线,单元格线宽为0.5,颜色为黑色,使每个单元格分离更加明显,数据更容易读取。

16. 自定义显示数据格式

观察上述在Seaborn绘制的热力图中的小数,有的保留小数点后两位,有的为小数点后一位,看起来不太完美!

事实上,我们可以设置带四舍五入小数点的格式,一般通过在函数heatmap中使用fmt参数控制。此参数允许大家指定格式化输出的字符串来控制输出格式。

例如,如果要显示四舍五入到小数点后两位的数字,可以使用fmt=".2f"。代码如下:

import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
np.random.seed(0)
data = pd.DataFrame(np.random.randn(10, 10))
corr = data.corr()
plt.figure(figsize=(10, 8))
sns.heatmap(corr, annot=True, fmt=".2f", cmap='coolwarm')
plt.show()

运行后结果如下:

在这里插入图片描述

17. 总结

本文重点介绍了在Python中如何利用Seaborn库来绘制热力图,用来直观的可视化数据的密度和分布,并详细介绍了不同参数控制下的可视化效果,并给出了相应的代码示例!

您学废了嘛!

这篇关于一文弄懂Seaborn绘制热力图的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/912989

相关文章

一文详解Git中分支本地和远程删除的方法

《一文详解Git中分支本地和远程删除的方法》在使用Git进行版本控制的过程中,我们会创建多个分支来进行不同功能的开发,这就容易涉及到如何正确地删除本地分支和远程分支,下面我们就来看看相关的实现方法吧... 目录技术背景实现步骤删除本地分支删除远程www.chinasem.cn分支同步删除信息到其他机器示例步骤

使用Python绘制3D堆叠条形图全解析

《使用Python绘制3D堆叠条形图全解析》在数据可视化的工具箱里,3D图表总能带来眼前一亮的效果,本文就来和大家聊聊如何使用Python实现绘制3D堆叠条形图,感兴趣的小伙伴可以了解下... 目录为什么选择 3D 堆叠条形图代码实现:从数据到 3D 世界的搭建核心代码逐行解析细节优化应用场景:3D 堆叠图

一文详解Java Stream的sorted自定义排序

《一文详解JavaStream的sorted自定义排序》Javastream中的sorted方法是用于对流中的元素进行排序的方法,它可以接受一个comparator参数,用于指定排序规则,sorte... 目录一、sorted 操作的基础原理二、自定义排序的实现方式1. Comparator 接口的 Lam

一文深入详解Python的secrets模块

《一文深入详解Python的secrets模块》在构建涉及用户身份认证、权限管理、加密通信等系统时,开发者最不能忽视的一个问题就是“安全性”,Python在3.6版本中引入了专门面向安全用途的secr... 目录引言一、背景与动机:为什么需要 secrets 模块?二、secrets 模块的核心功能1. 基

一文详解MySQL如何设置自动备份任务

《一文详解MySQL如何设置自动备份任务》设置自动备份任务可以确保你的数据库定期备份,防止数据丢失,下面我们就来详细介绍一下如何使用Bash脚本和Cron任务在Linux系统上设置MySQL数据库的自... 目录1. 编写备份脚本1.1 创建并编辑备份脚本1.2 给予脚本执行权限2. 设置 Cron 任务2

一文详解如何在idea中快速搭建一个Spring Boot项目

《一文详解如何在idea中快速搭建一个SpringBoot项目》IntelliJIDEA作为Java开发者的‌首选IDE‌,深度集成SpringBoot支持,可一键生成项目骨架、智能配置依赖,这篇文... 目录前言1、创建项目名称2、勾选需要的依赖3、在setting中检查maven4、编写数据源5、开启热

一文全面详解Python变量作用域

《一文全面详解Python变量作用域》变量作用域是Python中非常重要的概念,它决定了在哪里可以访问变量,下面我将用通俗易懂的方式,结合代码示例和图表,带你全面了解Python变量作用域,需要的朋友... 目录一、什么是变量作用域?二、python的四种作用域作用域查找顺序图示三、各作用域详解1. 局部作

一文彻底搞懂Java 中的 SPI 是什么

《一文彻底搞懂Java中的SPI是什么》:本文主要介绍Java中的SPI是什么,本篇文章将通过经典题目、实战解析和面试官视角,帮助你从容应对“SPI”相关问题,赢得技术面试的加分项,需要的朋... 目录一、面试主题概述二、高频面试题汇总三、重点题目详解✅ 面试题1:Java 的 SPI 是什么?如何实现一个

QT6中绘制UI的两种方法详解与示例代码

《QT6中绘制UI的两种方法详解与示例代码》Qt6提供了两种主要的UI绘制技术:​​QML(QtMeta-ObjectLanguage)​​和​​C++Widgets​​,这两种技术各有优势,适用于不... 目录一、QML 技术详解1.1 QML 简介1.2 QML 的核心概念1.3 QML 示例:简单按钮

一文详解PostgreSQL复制参数

《一文详解PostgreSQL复制参数》PostgreSQL作为一款功能强大的开源关系型数据库,其复制功能对于构建高可用性系统至关重要,本文给大家详细介绍了PostgreSQL的复制参数,需要的朋友可... 目录一、复制参数基础概念二、核心复制参数深度解析1. max_wal_seChina编程nders:WAL