[英]Seaborn heatmap total sum
嗨,我有一個 seaborn 熱圖,我想在右側添加每行的總和,並在熱圖底部添加每列的總和。 這可能嗎?
confusion_matrix = ([[417924, 67554],
[ 24901, 11070]])
x_axis_labels = ['predicted_non_churns','predicted_churns'] # labels for x-axis
y_axis_labels = ['actual_non_churns','actual_churns'] # labels for y-axis
ax = plt.axes()
ax.set_title('Confusion Matrix',fontsize=14, fontweight='bold')
sn.heatmap(confusion_matrix, annot=True, cmap="Purples",
xticklabels=x_axis_labels, yticklabels=y_axis_labels, fmt='g', ax=ax) # font size
您可以沿軸取混淆矩陣的總和,並使用 numpy 的堆疊函數將其堆疊到自身上。
import numpy as np
import seaborn as sn
import matplotlib.pyplot as plt
confusion_matrix = np.array([[417924, 67554],
[ 24901, 11070]])
confusion_matrix = np.hstack((confusion_matrix,confusion_matrix.sum(axis=1).reshape(-1,1)))
confusion_matrix = np.vstack((confusion_matrix,confusion_matrix.sum(axis=0).reshape(1,-1)))
x_axis_labels = ['predicted_non_churns','predicted_churns','col1+col2'] # labels for x-axis
y_axis_labels = ['actual_non_churns','actual_churns','row1+row2'] # labels for y-axis
ax = plt.axes()
ax.set_title('Confusion Matrix',fontsize=14, fontweight='bold')
sn.heatmap(confusion_matrix, annot=True, cmap="Purples",
xticklabels=x_axis_labels, yticklabels=y_axis_labels, fmt='g', ax=ax) # font size
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.