Python对Excel生成图表并保存至Excel

一、整体流程

flowchart TD
    A(读取Excel数据) --> B(生成图表数据)
    B --> C(绘制图表)
    C --> D(保存至Excel)

二、具体步骤及代码

1. 读取Excel数据

import pandas as pd

# 读取Excel文件
data = pd.read_excel('data.xlsx')

2. 生成图表数据

import matplotlib.pyplot as plt

# 设置图表数据
x = data['x']
y = data['y']

3. 绘制图表

# 绘制折线图
plt.plot(x, y)
plt.xlabel('X label')
plt.ylabel('Y label')
plt.title('Title')
plt.show()

4. 保存至Excel

# 保存图表至Excel
fig = plt.figure()
plt.plot(x, y)
plt.xlabel('X label')
plt.ylabel('Y label')
plt.title('Title')
fig.savefig('chart.png')

# 将图片插入Excel
from openpyxl import Workbook
from openpyxl.drawing.image import Image

wb = Workbook()
ws = wb.active
img = Image('chart.png')
ws.add_image(img, 'A1')
wb.save('output.xlsx')

三、类图

classDiagram
    class ExcelData{
        + read_excel()
    }
    class ChartData{
        + generate_data()
    }
    class DrawChart{
        + draw_chart()
    }
    class SaveToExcel{
        + save_to_excel()
    }

通过以上步骤，你就可以实现Python对Excel生成图表并保存至Excel的功能了。希望对你有所帮助，加油！