如何在statsmodel马赛克图中调整颜色和文本？

import matplotlib.pyplot as plt
import numpy as np

# 创建一个示例数据
categories = ['A', 'B', 'C', 'D']
data = np.array([[10, 20, 30, 40],
                 [15, 25, 35, 45],
                 [20, 30, 40, 50]])

# 计算每个分类的总和
category_totals = data.sum(axis=1)

# 计算每个分类的相对频率
relative_frequencies = data / category_totals[:, np.newaxis]

# 创建马赛克图
mosaic = plt.pcolormesh(relative_frequencies, cmap='Blues')

# 添加颜色条
plt.colorbar(mosaic)

# 添加分类标签
for i in range(len(categories)):
    for j in range(len(data[i])):
        plt.text(j + 0.5, i + 0.5, data[i][j], ha='center', va='center')

# 调整坐标轴和标签
plt.xticks(np.arange(len(data[0])) + 0.5, range(len(data[0])))
plt.yticks(np.arange(len(data)) + 0.5, categories)
plt.xlabel('X Label')
plt.ylabel('Y Label')

# 显示图形
plt.show()