Python matlibplot:可视化具有圆(相同半径，不同颜色)的整数矩阵(-1，0或1)

import matplotlib.pyplot as plt
import numpy as np

# 定义整数矩阵
matrix = np.array([[1, 0, -1],
                   [0, 1, 0],
                   [-1, 0, 1]])

# 获取矩阵的行数和列数
rows, cols = matrix.shape

# 定义圆的半径
radius = 0.5

# 遍历整数矩阵，绘制圆
for i in range(rows):
    for j in range(cols):
        # 计算圆心坐标
        x = j
        y = rows - i - 1
        
        # 根据整数的值确定圆的颜色
        if matrix[i, j] == -1:
            color = 'red'
        elif matrix[i, j] == 0:
            color = 'green'
        else:
            color = 'blue'
        
        # 绘制圆
        circle = plt.Circle((x, y), radius, color=color)
        plt.gca().add_patch(circle)

# 设置坐标轴范围
plt.xlim(-0.5, cols - 0.5)
plt.ylim(-0.5, rows - 0.5)

# 隐藏坐标轴
plt.axis('off')

# 显示图形
plt.show()