一、数据准备
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。还有其他颜色自行查找哦~
原文标题 : 一行代码实现“数据→热力图”