使用seaborn在python中将字典值映射为热图中的标签

2024-09-29 05:17:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我用Seaborn制作了一张热图。我有一个数据框“all_data_2018”,下面的代码创建了一个矩阵,看起来几乎完全符合我的要求

fig, ax = plt.subplots(figsize=(20,10))
title = 'Correlation Heatmap'
plt.title(title,fontsize=22)
ttl = ax.title
ttl.set_position([0.5,1.05])
ax.set_xticks([])
ax.set_yticks([])

sns.heatmap(all_data_2018.corr(),annot=True,fmt='.1f',cmap='RdYlGn',linewidths=0.2,ax=ax)
plt.show()

Screenshot of current correlation matrix

我遇到的问题是,标签是系列ID,我更希望它们是标签。我有一本字典:

my_dict = {'TM.VAL.INSF.ZS.WT':'insfs','SP.POP.TOTL.FE.ZS':'fempop', 'SP.DYN.IMRT.IN':'infmort', 'SN.ITK.MSFI.ZS':'foodinsec', 'SI.DST.FRST.10':'inclow10', 'SI.DST.10TH.10':'inchigh10','SP.ADO.TFRT':'adolefert', 'SL.UEM.NEET.FE.ZS':'femneec', 'SL.UEM.NEET.MA.ZS':'maleneec', 'SG.LAW.INDX':'wbli', 'SG.GEN.PARL.ZS':'womparl', 'SE.TER.TCHR.FE.ZS':'femtereduc', 'NY.GDP.PCAP.KD':'gdppercap'}

是否有一种方法可以轻松地将相关矩阵中的标签从本词典的键替换为值?我希望有一种方法可以很容易地绘制它,但我正在努力在任何地方找到它

谢谢


Tags: datatitleplt标签allaxspset