python绘制已知数值的混淆矩阵

import numpy as np

import pandas as pd

import seaborn as sns

import matplotlib.pyplot as plt

 

# 设置图形大小

plt.figure(figsize=(8,7))

 

# 创建一个3*3的混淆矩阵，已知其对应位置上的数字

cm = np.array([[1, 0, 0],

               [0, 23, 9],

               [0, 4, 3]])

 

xtick=['chat','video','web']

ytick=['chat','video','web']

# 将混淆矩阵转换为数据框，并添加行列标签

cm_df = pd.DataFrame(cm,

                     index=['chat','video','web'],

                     columns=['chat','video','web'])

sns.set(font_scale=1)

# sns.set(font_scale=10)

# 使用seaborn库画热力图，并设置相关参数

sns.heatmap(cm_df,

            cmap='Greys',

            cbar=False,

            xticklabels=xtick,