当我使用多索引数据框时，如何在seaborn heatmap中自定义y标签？

 admin  2023-05-24
Share: Twitter Facebook Pinterest Reddit VK Digg Linkedin Mix

我有一个具有多个索引行的DataFrame，我想创建一个没有重复行标签的热图，就像它在pandas DataFrame中出现的那样。下面是复制我的问题的代码：

import pandas as pd
from matplotlib import pyplot as plt
import random
import seaborn as sns
%matplotlib inline

df = pd.DataFrame({'Occupation':['Economist','Economist','Economist','Engineer','Engineer','Engineer',
                           'Data Scientist','Data Scientist','Data Scientist'],
             'Sex':['Female','Male','Both']*3, 'UK':random.sample(range(-10,10),9),
             'US':random.sample(range(-10,10),9),'Brazil':random.sample(range(-10,10),9)})

df = df.set_index(['Occupation','Sex'])

df

?

?

sns.heatmap(df, annot=True, fmt="",cmap="YlGnBu")

?

?

除了消除重复之外，我还想自定义一些y-labels，因为这个原始表单对我来说不太好。

有可能吗？

转载请注明出处:http://www.shenzhenzsmy.com/article/20230524/1551173.html