seaborn中常用的最喜欢画的图heatmap热度图

完整笔记+可视化图链接地址

#heatmap热度图,seaborn中常用的图,也是我最喜欢画的一种图

#重要点思维:拿到一批数据一般会求特征之间的相关系数,可以用padas直接求出来相关系数,放到heatmap,可以很清楚的看到两个特征的相关程度,这是一个固定的数据思维

#用途:比如拿到一批离散数据,想看一下在哪个点值比较大,在哪个点值比较低,你想把这样一个值的变化,用颜色来区分出来,这是我们要做的一个变化

import matplotlib.pyplot as plt

import numpy as np

np.random.seed(0)

import seaborn as sns

sns.set()

#颜色可以由浅到深,可以由深到浅,比如股票每天的涨跌

#

#随机生成一个3*3矩阵,点heatmap穿进去数据,调色板叫做col_bar,很明显的看出这堆数中的值大小

uniform_data=np.random.rand(3,3)

heatmap=sns.heatmap(uniform_data)

#可以区间设置,vmin vmax,大于或小于v的全是一个颜色,只有在这区间的才会分颜色

ax=sns.heatmap(uniform_data,vmin=0.2,vmax=0.5)

#比如拿到的数据是权重参数,又有正负,正是涨,负是跌,定义center=0,以0为中心画这个数据

normal_data = np.random.randn(3, 3)

ax2 = sns.heatmap(normal_data, center=0)

#读取航班数据集flights,seaborn自带的,1949年乘机的人数passengers

#需要横轴表示年份,纵轴月份,点的值是大小

# 把当前的数据转换为可以用的矩阵格式,读取的dataframe的,然后.pivot一下(x,y,值)x和y直接写列名即可,直接把dataframe中的year和month传进来,加一个注释项annot=True,fmt=“d”即是在图上显示数据值,linwidth=.5加上一个格,这个图会比较更清晰,调色板是cmap=“YIGnBu”,颜色,cbar=false是隐藏,但是一般不隐藏不然不知道图例了

#默认颜色太丑,应该设置一下常用的颜色

flights = sns.load_dataset('flights')

# print(flights.head())

flights=flights.pivot('month','year','passengers') #pivot函数重要

# print(flights.head())

sns.heatmap(flights) #注意这里是直接传入数据集即可,不需要再单独传入x和y了

sns.heatmap(flights,linewidth=.5,annot=True,fmt='d')

#改变颜色

ax= sns.heatmap(flights,cbar=False,cmap='YlGnBu')

你可能感兴趣的:(seaborn中常用的最喜欢画的图heatmap热度图)