200722-Seaborn绘制热力混淆矩阵自定义刻度_ax=sns.heatmap cba的刻度-CSDN博客

本文链接：https://blog.csdn.net/qq_33039859/article/details/107527676

本文介绍如何使用Python的sklearn、matplotlib、seaborn等库绘制混淆矩阵和数据热力图，通过实例展示了混淆矩阵和热力图的生成过程，包括数据准备、图表配置及展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

sns.set()

y_test = np.random.randint(1,8,100)
y_pred = np.random.randint(1,8,100)


fig = plt.figure()
ax = fig.add_subplot(111)
conf_mat = confusion_matrix(y_test, y_pred)
sns.heatmap(conf_mat,annot=True,xticklabels=['0','1','2','3','4','5','8'],yticklabels=['0','1','2','3','4','5','8']) #画热力图
ax.set_title('confusion matrix') #标题
ax.set_xlabel('predict') #x轴
ax.set_ylabel('true') #y轴
plt.show()

在这里插入图片描述

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import matplotlib.pyplot as plt



data = [[89.65, 90.47, 87.99, 89.51, 89.13, 88.85],
        [89.41, 88.70, 86.06, 89.81, 89.79, 89.73],
        [89.38, 90.72, 87.40, 89.91, 91.30, 89.68],
        [89.04, 89.51, 89.46, 88.81, 89.63, 89.39],
        [90.35, 88.36, 89.66, 88.67, 90.46, 89.21],
        [89.90, 89.94, 90.94, 83.64, 89.27, 90.79]]
data = np.array(data)
xticks = [1,4,16,64,26,1024]
yticks = [1,4,16,64,26,1024]


fig, ax = plt.subplots(figsize=(6,4)) 
ax = sns.heatmap(data, 
                cmap="YlGnBu", 
                annot=True, 
                fmt=".2f", 
                linewidths=2, 
                square=True,
                xticklabels=xticks,
                yticklabels=yticks,)

ax.set_xlabel('Target number $\ell_T$', family='Arial')
ax.set_ylabel('Source number $\ell_S$', family='Arial')
ax.set_title('Max:91.30, Min:83.64, Mean:89.29, Std:1.37.', family='Arial')
plt.tight_layout()
plt.savefig('Heatmap.png', dpi=300)