百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

如何使用 Python 将图表写入 Excel

itomcoil 2025-08-01 17:44 5 浏览

将 Python 生成的图表写入 Excel 文件是数据分析和可视化中常见的需求。Python 提供了多种库(如 matplotlib、openpyxl 和 xlsxwriter)来实现这一功能。本文将详细介绍如何使用 Python 创建图表并将其嵌入 Excel 文件,涵盖所需工具、代码示例以及注意事项。


1. 准备工作

在开始之前,需要安装以下 Python 库:

  • matplotlib:用于生成图表。
  • openpyxl:用于操作 Excel 文件,支持将图表嵌入。
  • xlsxwriter:另一种操作 Excel 文件的库,适合生成复杂的图表。
  • pandas:用于数据处理,便于生成 Excel 文件中的表格数据。

可以通过以下命令安装这些库:

pip install matplotlib openpyxl xlsxwriter pandas

此外,确保系统中已安装 Microsoft Excel 或兼容的软件(如 LibreOffice)以查看生成的 Excel 文件。


2. 使用xlsxwriter 将图表写入 Excel

xlsxwriter 是一个功能强大的库,支持直接在 Excel 文件中创建图表。以下是一个示例,展示如何生成一个折线图并将其写入 Excel 文件。

示例代码:生成折线图并写入 Excel

import pandas as pd
import xlsxwriter

# 创建示例数据
data = {
    'Month': ['Jan', 'Feb', 'Mar', 'Apr', 'May'],
    'Sales': [200, 240, 270, 300, 320]
}
df = pd.DataFrame(data)

# 创建 Excel 文件
workbook = xlsxwriter.Workbook('chart_example.xlsx')
worksheet = workbook.add_worksheet()

# 写入数据到 Excel
for row_num, (month, sales) in enumerate(zip(df['Month'], df['Sales'])):
    worksheet.write(row_num + 1, 0, month)  # 写入月份
    worksheet.write(row_num + 1, 1, sales)  # 写入销量

# 创建折线图
chart = workbook.add_chart({'type': 'line'})

# 配置图表数据
chart.add_series({
    'name': 'Sales Data',
    'categories': '=Sheet1!$A$2:$A$6',  # X轴数据范围
    'values': '=Sheet1!$B$2:$B$6',      # Y轴数据范围
})

# 设置图表标题和轴标签
chart.set_title({'name': 'Monthly Sales'})
chart.set_x_axis({'name': 'Month'})
chart.set_y_axis({'name': 'Sales'})

# 将图表插入工作表
worksheet.insert_chart('D2', chart)

# 关闭工作簿
workbook.close()

代码说明

  1. 数据准备:使用 pandas 创建一个包含月份和销量的简单数据集。
  2. 创建 Excel 文件:通过 xlsxwriter.Workbook 创建一个新的 Excel 文件,并添加一个工作表。
  3. 写入数据:将数据写入 Excel 表格的指定单元格。
  4. 创建图表:使用 add_chart 方法创建折线图,并通过 add_series 指定数据范围。
  5. 设置图表属性:设置标题、X 轴和 Y 轴的标签。
  6. 插入图表:将图表插入到工作表的指定位置(例如 'D2')。
  7. 保存文件:关闭工作簿,保存 Excel 文件。

运行代码后,将生成一个名为 chart_example.xlsx 的文件,包含数据表格和折线图。


3. 使用openpyxl 将matplotlib 图表写入 Excel

如果需要更复杂的图表(如 matplotlib 提供的多样化可视化),可以使用 openpyxl 将生成的图表图片嵌入 Excel 文件。以下是一个示例,展示如何生成柱状图并嵌入 Excel。

示例代码:生成柱状图并嵌入 Excel

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

# 创建示例数据
data = {
    'Month': ['Jan', 'Feb', 'Mar', 'Apr', 'May'],
    'Sales': [200, 240, 270, 300, 320]
}
df = pd.DataFrame(data)

# 创建柱状图
plt.figure(figsize=(6, 4))
plt.bar(df['Month'], df['Sales'], color='skyblue')
plt.title('Monthly Sales')
plt.xlabel('Month')
plt.ylabel('Sales')
plt.tight_layout()

# 将图表保存到内存中的字节流
buf = io.BytesIO()
plt.savefig(buf, format='png')
buf.seek(0)

# 创建 Excel 文件
wb = Workbook()
ws = wb.active
ws.title = 'Sales Data'

# 写入数据到 Excel
ws.append(['Month', 'Sales'])  # 写入表头
for row in df.itertuples(index=False):
    ws.append([row.Month, row.Sales])

# 将图表图片插入 Excel
img = Image(buf)
ws.add_image(img, 'D2')

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

# 关闭字节流
buf.close()

代码说明

  1. 生成图表:使用 matplotlib 创建一个柱状图,并设置标题、轴标签等。
  2. 保存到内存:将图表保存为 PNG 格式的字节流(io.BytesIO),避免生成临时文件。
  3. 创建 Excel 文件:使用 openpyxl.Workbook 创建 Excel 文件,并写入数据。
  4. 插入图片:通过 openpyxl.drawing.image.Image 将图表图片插入到指定单元格(例如 'D2')。
  5. 保存文件:保存生成的 Excel 文件。

运行代码后,将生成一个名为
chart_example_openpyxl.xlsx 的文件,包含数据表格和嵌入的柱状图。


4. 两种方法的比较

特性

xlsxwriter

openpyxl + matplotlib

图表类型

Excel 原生图表(折线图、柱状图等)

任何 matplotlib 支持的图表

图表自定义程度

受限于 Excel 图表功能

高,matplotlib 提供丰富自定义选项

文件大小

较小(原生图表)

较大(嵌入图片)

依赖性

仅需 xlsxwriter

需 matplotlib 和 openpyxl

适用场景

简单的 Excel 原生图表

复杂或高度自定义的图表

  • 如果需要生成标准的 Excel 图表(如折线图、柱状图),推荐使用 xlsxwriter,因为它直接生成 Excel 原生图表,文件更小且可编辑。
  • 如果需要复杂的可视化(如热图、3D 图形),推荐使用 matplotlib 结合 openpyxl,但需要注意图片嵌入会增加文件大小。

5. 注意事项

  1. 依赖管理:确保安装了所有必要的库,并检查版本兼容性。
  2. 文件路径:保存 Excel 文件时,确保路径有效且有写入权限。
  3. 图表大小:在 matplotlib 中设置适当的 figsize,以确保图表在 Excel 中显示清晰。
  4. 性能问题:嵌入大量图片可能会导致 Excel 文件变大,影响打开速度。
  5. 兼容性:生成的 Excel 文件应在 Microsoft Excel 或 LibreOffice 中测试,确保图表正确显示。

6. 扩展:批量生成图表

如果需要为多组数据生成多个图表,可以通过循环创建多个工作表或图表。例如,使用 xlsxwriter:

import pandas as pd
import xlsxwriter

# 示例数据
data = {
    'Category': ['A', 'B', 'C'],
    'Sales1': [200, 240, 270],
    'Sales2': [300, 280, 250]
}
df = pd.DataFrame(data)

# 创建 Excel 文件
workbook = xlsxwriter.Workbook('multi_chart_example.xlsx')

for i, col in enumerate(['Sales1', 'Sales2']):
    worksheet = workbook.add_worksheet(f'Sheet{i+1}')
    
    # 写入数据
    worksheet.write(0, 0, 'Category')
    worksheet.write(0, 1, col)
    for row_num, (cat, val) in enumerate(zip(df['Category'], df[col])):
        worksheet.write(row_num + 1, 0, cat)
        worksheet.write(row_num + 1, 1, val)
    
    # 创建折线图
    chart = workbook.add_chart({'type': 'line'})
    chart.add_series({
        'name': col,
        'categories': f'=Sheet{i+1}!$A$2:$A$4',
        'values': f'=Sheet{i+1}!$B$2:$B$4',
    })
    
    # 设置图表标题
    chart.set_title({'name': f'{col} by Category'})
    
    # 插入图表
    worksheet.insert_chart('D2', chart)

# 保存文件
workbook.close()

此代码为每组数据(Sales1 和 Sales2)生成一个工作表和对应的折线图。


7. 结论

通过 xlsxwriter 或 openpyxl 结合 matplotlib,可以将 Python 生成的图表轻松写入 Excel 文件。选择哪种方法取决于你的需求:如果追求简单和原生支持,选择 xlsxwriter;如果需要高度自定义的图表,选择 openpyxl 结合 matplotlib。另外,xlwings也提供了把图表写回EXCEL工作表的功能,大家感兴趣的话也可以去了解和探索。希望本文的代码和说明能帮助你顺利完成图表嵌入任务!

相关推荐

python学习教程-第七节内容

函数(重点)基本概念Python函数的语法是编程中的核心概念之一,它允许你将代码块封装为可重复调用的单元。基本语法定义函数:示例参数类型位置参数(PositionalArguments)按参数定义...

Python排序90%人只会用sorted()?这7个高阶技巧让你代码效率翻倍

高效处理数据的关键,往往从掌握排序开始。排序操作在Python编程中的重要性常被低估——直到你面对一个杂乱无章的数据集。作为数据处理的核心操作之一,排序效率直接决定了程序性能和代码可读性。无论你正在清...

第四章:Python函数

4.1函数的定义与调用4.1.1理论知识函数是组织好的、可重复使用的代码块,用于执行特定的任务。它可以提高代码的复用性和可维护性。在Python中,定义函数使用def关键字,其基本语法如下:def...

ArcGIS 一个独立运行的Python程序编写和打包

写代码#coding=utf8#-*-coding:UTF-8-*-importarcpyfromarcpyimportenvimportosimportsys##########...

python入门到脱坑经典案例—计算三角形的面积

下面为大家详细讲解如何用Python计算三角形面积。我们会从最基础的数学公式开始,逐步深入编程实现,并涵盖多个实用版本。以下是完整的教学指南:1.数学原理回顾三角形面积公式:面积=(底边长度...

Python运算技巧:10种计算列表平方的高效方法

1.使用for循环此方法遍历列表中的每个数字,使用**运算符计算其平方,然后将结果添加到新的列表中。2.使用列表推导式此方法使用列表推导式,这是一种更简洁的方式,可以在现有列表的每个项目上执行操作...

墙裂推荐,5个学习Python编程最佳的开源库

学习Python少不了的就是项目,我在GitHUB上找了几个比较好的开源库,大家可以跟着去学习。1、learn-python3这个存储库一共有19本Jupyter笔记本。它涵盖了字符串和条件之类的基础...

使用Python玩转ftplib库:实现ftp文件传输自动化全攻略

大家好!在日常工作中都会使用到ftp功能,用于上传和下载文件等,本章主要介绍Python的标准库ftplib来实现FTP文件传输,帮助我们实现ftp自动化。一、ftplib库核心函数速查表1连接与登...

Python零基础入门学习23:常用第三方库之图像处理库Pillow

注:本文所有代码均经过Python3.7实际运行检验,保证其严谨性。本文约2000字,阅读时间约为4分钟。Pillow库的概述Pillow库是Python最好的图像处理库,可能是使用频率最高的图像处...

Python编程之BeautifulSoup库

#头条创作挑战赛#BeautifulSoup是一个可以从HTML或XML文件中快速提取数据的Python库。它能够通过转从入门到精通Python网络爬虫框架Scrapy38换器实...

Python3 新一代Http请求库Httpx使用(详情版)

我们经常使用Python语言的朋友们都清楚,requests是使用率非常高的HTTP库,甚至更早Python2中使用的是urllib、urllib2,也给我们提供了许多方便的功能。但是自从...

小白也可以玩的Python爬虫库,收藏一下

最近,微软开源了一个项目叫「playwright-python」,作为一个兴起项目,出现后受到了大家热烈的欢迎,那它到底是什么样的存在呢?今天为你介绍一下这个传说中的小白神器。Playwright是...

apscheduler,一个超厉害的 Python 库!

大家好,今天为大家分享一个超厉害的Python库-apscheduler。Github地址:https://github.com/agronholm/apschedulerAPSchedule...

给3D Slicer添加Python第三方插件库

对临床医生或医学影像算法研究人员来说,3DSlicer是不错的临床实践或科研工具。随着人工智能和深度学习技术的普及,python已经作为原型开发和验证的计算机编程语言。3DSlicer自带pyt...

Paramiko:一个非常实用的Python库

前言Python的Paramiko库,它是一个用于实现SSHv2协议的客户端和服务器的库。通过使用Paramiko,我们可以在Python程序中轻松地实现远程服务器的管理、文件传输等功能。特别做智能硬...