在Python中绘制带有自定义x和y轴的混淆矩阵？

import numpy as np
import matplotlib.pyplot as plt

# 定义混淆矩阵数据
confusion_matrix = np.array([[10, 2, 3],
                             [4, 15, 6],
                             [7, 8, 20]])

# 定义自定义x和y轴标签
labels = ['Class 1', 'Class 2', 'Class 3']

# 绘制混淆矩阵
fig, ax = plt.subplots()
im = ax.imshow(confusion_matrix, cmap='Blues')

# 设置颜色条
cbar = ax.figure.colorbar(im, ax=ax)

# 设置x和y轴刻度
ax.set_xticks(np.arange(len(labels)))
ax.set_yticks(np.arange(len(labels)))
ax.set_xticklabels(labels)
ax.set_yticklabels(labels)

# 在矩阵格子中显示数值
for i in range(len(labels)):
    for j in range(len(labels)):
        text = ax.text(j, i, confusion_matrix[i, j],
                       ha='center', va='center', color='black')

# 设置图表标题和标签
ax.set_title('Confusion Matrix')
ax.set_xlabel('Predicted Label')
ax.set_ylabel('True Label')

# 显示图表
plt.show()