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

最新推荐文章于 2024-04-06 06:01:43 发布

兰振lanzhen

最新推荐文章于 2024-04-06 06:01:43 发布

阅读量532

点赞数 1

分类专栏：可视化文章标签： python excel

本文链接：https://blog.csdn.net/lanzhen001/article/details/131775931

版权

可视化专栏收录该内容

5 篇文章 1 订阅

订阅专栏

该文介绍了如何使用Python进行数据可视化，特别是将DataFrame数据转化为Numpy数组，然后通过Matplotlib库创建棋盘图。主要步骤包括数据转换、查找0和1的索引、绘制散点图以及定制坐标刻度显示文本，最后展示图表。

摘要由CSDN通过智能技术生成

文章目录

一、需求

根据可视化的需要，下图的数据需要使用棋盘图的样式来展示，
原始数据：
在这里插入图片描述
最终效果图：

二、处理方式

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()