python--seaborn热力图

FTDdata · 简书 · · 2021-05-05 09:02

正文

seaborn是专门用于统计数据可视化的包，可媲美R语言中的ggplot2包。本文介绍用seaborn绘制热力图。

环境

python3.9
win10 64bit
seaborn==0.11.1
matplotlib==3.3.4
pandas==1.2.1

数据

热力图的一个常见应用场景是绘制相关系数热力图，数据准备一个相关系数矩阵。

import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams['figure.dpi'] = 80  # 图形分辨率
pd.options.display.notebook_repr_html=False  # 表格显示

# 相关系数矩阵
np.random.seed(1)  # 随机种子
mat=pd.DataFrame(np.random.rand(3,6),columns=list('abcdef')).corr()
mat

          a         b         c         d         e         f
a  1.000000  0.297407 -0.610065 -0.908297 -0.997893 -0.991160
b  0.297407  1.000000 -0.937936  0.129262 -0.234835 -0.421449
c -0.610065 -0.937936  1.000000  0.222660  0.557370  0.709796
d -0.908297  0.129262  0.222660  1.000000  0.933525  0.844767
e -0.997893 -0.234835  0.557370  0.933525  1.000000  0.980463
f -0.991160 -0.421449  0.709796  0.844767  0.980463  1.000000

绘制

调用 heatmap 方法绘制热力图。

# 热力图
sns.heatmap(mat)
plt.show()

heatmap_6_0.png

调整调色板

设置 vmin 和 vmax 参数可以调整调色板的下限值和上限值。

# 设置调色板上下限值
sns.heatmap(mat,vmin=0,vmax=1)
plt.show()

请到「今天看啥」查看全文

推荐文章

西安头条 · 痛心！一对90后夫妻被困身亡

昨天

扬子晚报 · 3·15曝光问题，多地连夜查处！

昨天

贵阳日报 · 贵阳市市场监督管理局连夜部署检查央视“3·15”晚会曝光的“非标”电线电缆问题

昨天

人民日报 · 两大知名茶饮品牌被曝光！店员：我也不敢喝

昨天

澎湃新闻 · 315暗访｜网红足浴包细菌“爆表”，涉北京同仁堂、马应龙等

昨天

老北京城 · 北京为什么是全国最牛逼的城市！看完这条微信我懂了！！

7 年前

KnowYourself · “我也很想Ta，我们都一样”｜团体心理咨询：可能是你5月最特别的体验

7 年前

药智网 · 【聚焦】“三关”保障药品安全，医院要严把“终端关”

7 年前

每日经济新闻 · 很重要！车载空调八大使用误区，你都知道吗？

7 年前

中国企业家杂志 · 雷军：创业公司招人与留人的方法

7 年前