如何生成组之间成对共享的矩阵并将其可视化

import matplotlib.pyplot as plt
import numpy as np

# 定义组和共享关系
groups = ['组A', '组B', '组C', '组D']
share_matrix = np.array([[0, 1, 1, 0],
                         [1, 0, 1, 1],
                         [1, 1, 0, 1],
                         [0, 1, 1, 0]])

# 绘制共享矩阵
fig, ax = plt.subplots()
im = ax.imshow(share_matrix, cmap='Blues')

# 设置轴标签
ax.set_xticks(np.arange(len(groups)))
ax.set_yticks(np.arange(len(groups)))
ax.set_xticklabels(groups)
ax.set_yticklabels(groups)

# 在单元格中显示数值
for i in range(len(groups)):
    for j in range(len(groups)):
        text = ax.text(j, i, share_matrix[i, j], ha='center', va='center', color='black')

# 添加颜色条
cbar = ax.figure.colorbar(im, ax=ax)

# 添加标题
ax.set_title('组之间的共享矩阵')

# 显示图形
plt.show()