如何将Pandas生成的柱状图保存至原Excel对应工作表？

核心思路 #

要把生成的柱状图插入到原Excel的对应工作表中，需借助openpyxl库操作Excel文件（支持修改已有文件且不覆盖其他内容），结合matplotlib生成图表并转为内存对象，再插入到指定工作表的合适位置。

完整实现步骤 #

首先安装必要依赖：

pip install pandas matplotlib openpyxl

修改你的代码如下：

import pandas as pd
import matplotlib.pyplot as plt
from openpyxl import load_workbook
from io import BytesIO
from openpyxl.drawing.image import Image

# 替换为你的Excel文件实际路径
path = "your_excel_file.xlsx"

# 以可写模式加载原有Excel工作簿
wb = load_workbook(path)

for p, hc in enumerate(sheet2, 1):
    # 读取对应工作表的数据
    histo = pd.read_excel(path, sheet_name=f'Histogram{p}')
    
    # 生成柱状图并调整布局
    plt.figure(figsize=(10, 6))
    histo.plot(kind='bar')
    plt.xlabel("Probe Name")
    plt.ylabel("Fold Change")
    plt.tight_layout()  # 避免标签被截断
    
    # 将图表存入内存对象（无需生成本地文件）
    img_buffer = BytesIO()
    plt.savefig(img_buffer, format='png', dpi=100)
    img_buffer.seek(0)
    plt.close()  # 关闭图表释放内存
    
    # 获取目标工作表并插入图像
    target_sheet = wb[f'Histogram{p}']
    img = Image(img_buffer)
    img.anchor = 'A10'  # 设置图表插入的起始单元格，可根据数据行数调整
    target_sheet.add_image(img)

# 保存修改后的Excel文件
wb.save(path)

关键细节说明 #

• openpyxl的作用：作为Excel操作引擎，支持加载已有工作簿并修改，不会破坏其他未操作的工作表。
• 内存缓冲区BytesIO：避免生成本地临时图像文件，直接在内存中处理图表数据，提升效率。
• 图像位置调整：通过img.anchor指定图表插入的起始单元格（如A10），可根据工作表内已有数据的行数调整，防止覆盖原有内容。
• 资源释放：每次生成图表后调用plt.close()，避免内存泄漏。

内容的提问来源于stack exchange，提问作者dhananya s