一行代码实现“数据→热力图”

2022-03-25 11:35

佐佑思维

关注

一、数据准备

import numpy as np

import seaborn as sns

df ＝ pd．DataFrame（np．array（data），

index＝［＇0＇，＇1＇，＇2＇，＇3＇，＇4＇，＇5＇，＇6＇，＇7＇，＇8＇］，

columns＝［＂P＂，＂Q＂，＂QN＂，＂TE＂，＂R＂，＂DE＂，＂S＂，＂WD＂，＂W＂］）

将你的数据导入为一个数组或列表即可；

在index和columns中输入你想建立的横纵坐标；

如果是一个excel可以直接参考这个文章，读取数据为一个pd格式数据。

二、绘图准备

import matplotlib．pyplot as plt

plt．figure（figsize ＝（12，6））

三、一行绘制热力图

ax ＝ sns．heatmap（df， annot＝True， cmap＝＂BuPu＂）＃cmap有很多可选的“色号”，如：YlGnBu，YlGnBu＿r，hot，hot＿r，OrRd，autumn，greens，viridis，greys，Purples，rainbow，gist＿rainbow。还有其他颜色自行查找哦～