当前位置： 技术文章>> Python 中如何操作 Excel 的图表数据？

文章标题：Python 中如何操作 Excel 的图表数据？
  • 文章分类: 后端
  • 7123 阅读

在Python中操作Excel图表数据，通常涉及到读取、修改以及可能的话，创建或更新Excel文件中的图表。虽然Python标准库并不直接支持Excel图表的复杂操作，但我们可以借助一些强大的第三方库来实现这些功能，其中openpyxl和pandas结合matplotlib或xlsxwriter是常用的解决方案。下面，我将详细介绍如何使用这些工具来操作Excel图表数据。

1. 使用openpyxl读取和修改Excel图表

openpyxl是一个用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。虽然openpyxl主要关注于工作表数据的操作，但它也提供了一些基本的图表处理能力，尤其是读取和修改现有图表。

安装openpyxl

首先，确保安装了openpyxl库。你可以通过pip安装：

pip install openpyxl

读取Excel图表

openpyxl允许你访问和修改Excel文件中的图表，但主要限于读取图表的基本信息和一些简单的修改。

from openpyxl import load_workbook

# 加载现有的xlsx文件
wb = load_workbook(filename='example.xlsx')
ws = wb.active  # 激活工作表

# 访问图表
charts = ws._charts  # 注意：这是一个内部属性，可能在未来的版本中有所变化
for chart in charts:
    print(chart.title.text)  # 打印图表的标题
    # 你可以进一步探索chart对象的属性和方法

# 保存修改（如果有的话）
wb.save('modified_example.xlsx')

注意：

  • 图表操作在openpyxl中相对有限，主要用于读取和简单修改。
  • 复杂的图表创建或修改可能需要使用其他工具。

2. 使用pandas和matplotlib生成图表并保存为Excel

虽然pandas和matplotlib不直接操作Excel文件中的图表，但你可以使用它们生成图表，然后将图表保存为图片，最后将这个图片作为对象插入到Excel文件中。这通常用于生成全新的图表。

安装必要的库

pip install pandas matplotlib openpyxl

使用pandas和matplotlib生成图表

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {'Year': [2018, 2019, 2020, 2021], 'Sales': [100, 150, 75, 200]}
df = pd.DataFrame(data)

# 使用matplotlib绘制图表
plt.figure(figsize=(8, 6))
plt.plot(df['Year'], df['Sales'], marker='o')
plt.title('Sales Over Years')
plt.xlabel('Year')
plt.ylabel('Sales')
plt.grid(True)

# 保存图表为图片
plt.savefig('sales_chart.png')
plt.close()

# 注意：这里还没有将图片插入Excel

将图片插入Excel

将图片插入Excel需要回到openpyxl，使用其添加图片的功能。

from openpyxl import Workbook
from openpyxl.drawing.image import Image

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

# 插入图片
img = Image('sales_chart.png')
ws.add_image(img, 'A1')  # 将图片添加到A1单元格的位置

# 保存工作簿
wb.save('chart_in_excel.xlsx')

3. 使用xlsxwriter创建图表

xlsxwriter是另一个强大的库，用于创建新的Excel文件，包括图表。它不支持读取或修改现有的Excel文件，但非常适合从头开始创建包含图表的Excel报告。

安装xlsxwriter

pip install xlsxwriter

使用xlsxwriter创建图表

import xlsxwriter

# 创建一个新的Excel文件和一个工作表
workbook = xlsxwriter.Workbook('chart_example.xlsx')
worksheet = workbook.add_worksheet()

# 准备一些数据
data = [
    ['Year', 'Sales'],
    [2018, 100],
    [2019, 150],
    [2020, 75],
    [2021, 200],
]

# 写入数据
row = 0
col = 0
for item in data:
    worksheet.write_row(row, col, item)
    row += 1

# 创建一个图表对象
chart = workbook.add_chart({'type': 'line'})

# 配置图表的数据系列
chart.add_series({
    'name':       '=Sheet1!$B$1',
    'categories': '=Sheet1!$A$2:$A$5',
    'values':     '=Sheet1!$B$2:$B$5',
})

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

# 关闭并保存文件
workbook.close()

总结

在Python中操作Excel图表数据，你可以根据需求选择openpyxl、pandas结合matplotlib或xlsxwriter。openpyxl适合读取和简单修改现有Excel文件中的图表，而pandas和matplotlib则擅长生成图表并将其作为图片插入Excel。xlsxwriter则专注于创建全新的Excel文件，包括复杂的图表。根据你的具体需求，选择最合适的工具组合来高效完成任务。

在实际应用中，你可能还会遇到需要将生成的图表或修改后的Excel文件上传至网站或进行进一步处理的情况。这时，你可以考虑将文件保存到服务器上的指定位置，或者使用Python的HTTP库（如requests）将文件作为附件发送到Web服务。对于码小课这样的网站，如果它提供了文件上传或API接口来接收文件，你可以轻松地将这些操作集成到你的Python脚本中，实现自动化处理。

推荐文章