在 seaborn 熱圖混淆矩陣中標記單元格的百分比和絕對值

Question

我嘗試在熱圖的單元格上同時寫入百分比和絕對值，例如“15 (20%)”。 我可以通過下面的代碼計算每個 class 的百分比。 但是，我怎樣才能為兩者做到這一點？

h_map = sns.heatmap(pyArray_cf / np.sum(pyArray_cf, axis=1),  vmin=0, vmax=1, annot=True, cmap="Blues",
                            ax=ax1, fmt='.2%',
                            mask=pyArray_cf==0,
                            xticklabels=x_axis_labels_1,
                            yticklabels=y_axis_labels_1)

另外，我想用 0 值（0 次混淆）來掩蓋單元格。

Answer 1

感謝特倫頓評論中提到的一些要點，我已經建立了下面的熱圖。 希望這是您正在尋找的。

• 使用div獲取每個單元格的百分比。 通過將每個單元格除以整個列的總和來計算百分比。 假設這是你需要的
• 通過轉換絕對值（小數點后 2 位）后跟換行符 (\n) 和百分比（小數點后 1 位）以及末尾的 % 符號來為每個單元格創建注釋
• 對每個單元格中的文本使用 annot。 在創建 plot 時調整大小和線條

data = np.random.rand(10, 12) #Random data 0 to 1
df = pd.DataFrame(data)

#Create perc to store the percentages for each column using DIV.
perc = df.copy()
cols=perc.columns.values
perc[cols]=perc[cols].div(perc[cols].sum(axis=1), axis=0).multiply(100)

#Use annot to store the text value to display in each cell. Number \n Percent % format
annot=df.round(2).astype(str) + "\n" + perc.round(1).astype(str) + "%"

#Create plot - adjust size, linewidth and provide annotation text for display
fig, ax = plt.subplots(figsize=(10,8))
sns.heatmap(data, annot=annot, fmt='', vmin=0, vmax=1, cmap="Blues", 
            annot_kws={"fontsize":8}, linewidth=1, ax=ax)

Plot