Python实现基于Excel数据绘制棋盘图
作者：兰振lanzhen
这篇文章主要为大家介绍了如何根据可视化的需要，利用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()
到此这篇关于Python实现基于Excel数据绘制棋盘图的文章就介绍到这了,更多相关Python棋盘图内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！