文章目录
- 一、需求
- 二、处理方式
- 三、代码实现
一、需求
根据可视化的需要,下图的数据需要使用棋盘图的样式来展示,
原始数据:
最终效果图:
二、处理方式
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()