如何绘制一个看起来像散点的0-1二维矩阵？

import numpy as np
import matplotlib.pyplot as plt

# 创建一个5x5的二维矩阵
matrix = np.random.randint(2, size=(5, 5))

# 获取矩阵的行数和列数
n, m = matrix.shape

# 获取矩阵中的0和1的坐标
zeros = np.argwhere(matrix == 0)
ones = np.argwhere(matrix == 1)

# 绘制散点图
plt.scatter(zeros[:, 1], zeros[:, 0], c='blue', label='0')
plt.scatter(ones[:, 1], ones[:, 0], c='red', label='1')

# 设置坐标轴和标题
plt.xlabel('Column')
plt.ylabel('Row')
plt.title('Scatter Plot of 0-1 Matrix')

# 显示图例
plt.legend()

# 显示散点图
plt.show()