我想创建一个热图,其中Y轴是树的数量,X轴是叶子的数量,在中心位置展示auc-roc。
以下是我的代码:
df = pd.DataFrame(store,columns = ['n_trees' , 'n_leafs', 'auc-roc'])
df.set_index(['n_trees'], inplace=True)
ax = sns.heatmap(df)
我的DataFrame如下:
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
我得到了这个热图,但不是我想要的。如何删除X轴上的auc-roc并将其转移到中心位置?