Comment construire une carte thermique ?

Question

Je veux construire une carte thermique où sur l'axe Y sera le nombre d'arbres, sur X le nombre de feuilles, et au centre l'auc-roc. Voici mon code

df = pd.DataFrame(store,columns = ['n_trees' , 'n_leafs', 'auc-roc']) 
df.set_index(['n_trees'], inplace=True)
ax = sns.heatmap(df)

Mon cadre de données ressemble à ceci :

         n_leafs   auc-roc
n_trees                   
10             1  0.7
10             3  0.892529
10             5  0.107495
159            1  0.155
159            3  0.7581
...          ...       ...
1202           3  0.420
1202           5  0.422
1351           1  0.398
1351           3  0.273
1351           5  0.795

et j'obtiens cette carte thermique, pas ce que je voulais. Comment supprimer l'auc-roc sur l'axe X et le transférer au centre ?