python根据excel数据，基于散点图绘制棋盘图

一、需求

根据可视化的需要，下图的数据需要使用棋盘图的样式来展示，
原始数据：

最终效果图：

二、处理方式

1、先将DataFrame数据转换为Numpy数组；
2、先使用np.transpose函数，找到0和1值的索引；
3、然后创建散点图；
4、完成散点图后，由于需要展示的坐标值是文本，所以在绘制坐标刻度时，plt.xticks和plt.yticks要使用第二个参数传递，使其显示的是文本。

三、代码实现

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

# 提供的 DataFrame 数据
data = pd.read_excel('demo.xlsx',index_col=0)

# 将 DataFrame 转换为 NumPy 数组
data_array = data.values

# 找到值为 0 和 1 的索引
zero_indices = np.transpose(np.nonzero(data_array == 0))
one_indices = np.transpose(np.nonzero(data_array == 1))

# 创建散点图
plt.scatter(one_indices[:, 1] + 1, one_indices[:, 0] + 1, c='black', marker='o')

# 修改刻度值显示值
plt.xticks(np.arange(1, data.shape[1]+1),data.columns)
plt.yticks(np.arange(1, data.shape[0]+1),data.index)

# 绘制网格线
plt.grid(True,which='both')

# 设置图表标题和轴标签
plt.title('Checkerboard Diagram')
plt.xlabel('X')
plt.ylabel('Y')

# 显示图例
plt.legend()

# 显示图表
plt.show()