Python插入Excel二维坐标并用Matplotlib绘图

Python插入Excel二维坐标并用Matplotlib绘图

本文将介绍如何使用Python将16个和150个二维坐标点数据插入Excel，并利用Matplotlib库绘制散点图。

import openpyxl
import matplotlib.pyplot as plt

# 创建一个新的Excel工作簿
wb = openpyxl.Workbook()
ws = wb.active

# 插入16个二维坐标点数据
coordinates_16 = [(1, 2), (3, 4), (5, 6), (7, 8), (9, 10), (11, 12), (13, 14), (15, 16),
                  (17, 18), (19, 20), (21, 22), (23, 24), (25, 26), (27, 28), (29, 30), (31, 32)]

ws.append(['X', 'Y'])
for coord in coordinates_16:
    ws.append(coord)

# 插入150个二维坐标点数据
coordinates_150 = []  # 假设这里有150个坐标点
for i in range(150):
    coordinates_150.append((i+1, i+1))

ws.append(['X', 'Y'])
for coord in coordinates_150:
    ws.append(coord)

# 保存Excel文件
wb.save('coordinates.xlsx')

# 绘制16个二维坐标点的图表
x_16 = [coord[0] for coord in coordinates_16]
y_16 = [coord[1] for coord in coordinates_16]

plt.figure()
plt.scatter(x_16, y_16)
plt.title('16 Coordinates')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

# 绘制150个二维坐标点的图表
x_150 = [coord[0] for coord in coordinates_150]
y_150 = [coord[1] for coord in coordinates_150]

plt.figure()
plt.scatter(x_150, y_150)
plt.title('150 Coordinates')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

代码解析：

1. 导入库: 首先导入 openpyxl 用于操作Excel文件，matplotlib.pyplot 用于绘图。
2. 创建Excel文件: 使用 openpyxl.Workbook() 创建一个新的Excel工作簿，并使用 wb.active 获取当前活动的工作表。
3. 插入坐标数据:
   • 定义了两个坐标列表 coordinates_16 和 coordinates_150。
   • 使用循环遍历坐标列表，并将每个坐标点写入Excel表格对应单元格。
4. 保存Excel文件: 使用 wb.save('coordinates.xlsx') 保存文件。
5. 绘制图表:
   • 使用列表推导式分别提取 x 和 y 坐标数据。
   • 使用 plt.scatter(x, y) 绘制散点图。
   • 使用 plt.title, plt.xlabel, plt.ylabel 设置图表标题和轴标签。
   • 使用 plt.show() 显示图表。

总结

本文介绍了使用Python将二维坐标点数据插入Excel并绘制散点图的方法，你可以根据实际需求修改代码，例如修改坐标数据、图表样式等。