python如何操作excel图表

在Python中操作Excel图表，主要有几种方法：使用库如openpyxl、通过pandas结合matplotlib生成图表、借助xlwings与Excel直接交互。其中，openpyxl和xlwings能直接操作Excel文件，而pandas与matplotlib则是用于数据处理和生成图表，最终再导入到Excel中。下面将详细介绍如何使用这些方法。

一、使用openpyxl操作Excel图表

Openpyxl 是一个用于读取和写入Excel 2010 xlsx/xlsm/xltx/xltm 文件的Python库。虽然它对图表支持有限，但仍可以创建基本的图表。

1. 创建简单图表

openpyxl允许创建一些基础的图表，如折线图、柱状图等。首先需要安装openpyxl：

pip install openpyxl

然后，可以使用以下代码创建一个简单的折线图：

from openpyxl import Workbook
from openpyxl.chart import LineChart, Reference
创建工作簿和工作表
wb = Workbook()
ws = wb.active
添加数据
rows = [
    ['Month', '2019', '2020'],
    ['Jan', 50, 30],
    ['Feb', 40, 25],
    ['Mar', 30, 40],
    ['Apr', 25, 35],
]
for row in rows:
    ws.append(row)
创建折线图
chart = LineChart()
data = Reference(ws, min_col=2, min_row=1, max_col=3, max_row=5)
chart.add_data(data, titles_from_data=True)
ws.add_chart(chart, "E5")
保存文件
wb.save("line_chart.xlsx")

2. 限制与缺点

Openpyxl对图表的支持虽然可以满足一些基本需求，但其功能相对有限，无法创建复杂的图表或进行太多自定义设置。如果需要实现更复杂的图表，可能需要借助其他工具。

二、使用pandas和matplotlib生成图表

Pandas和Matplotlib是两个非常强大的数据处理和可视化库，虽然它们不能直接操作Excel图表，但可以生成图表并将其导入到Excel中。

1. 安装和基础使用

首先，需要安装pandas和matplotlib：

pip install pandas matplotlib

然后，可以使用pandas读取数据，并使用matplotlib生成图表：

import pandas as pd
import matplotlib.pyplot as plt
创建数据框
data = {'Month': ['Jan', 'Feb', 'Mar', 'Apr'],
        '2019': [50, 40, 30, 25],
        '2020': [30, 25, 40, 35]}
df = pd.DataFrame(data)
设置月份为索引
df.set_index('Month', inplace=True)
绘制折线图
df.plot(kind='line')
plt.title('Sales Data')
plt.ylabel('Sales')
plt.xlabel('Month')
plt.savefig('sales_chart.png')
plt.show()

2. 将生成的图表导入Excel

生成图表后，可以使用openpyxl将其插入到Excel文件中：

from openpyxl import Workbook
from openpyxl.drawing.image import Image
创建工作簿和工作表
wb = Workbook()
ws = wb.active
插入图像
img = Image('sales_chart.png')
ws.add_image(img, 'E5')
保存文件
wb.save("sales_data.xlsx")

三、使用xlwings与Excel直接交互

Xlwings 是一个可以让Python与Excel无缝集成的库，适合在Excel中进行复杂的操作。它允许直接使用Excel的功能，包括图表的创建和管理。

1. 安装和基础使用

首先安装xlwings：

pip install xlwings

使用xlwings可以直接从Excel中读取数据，创建图表：

import xlwings as xw
启动Excel应用
app = xw.App(visible=True)
创建工作簿和工作表
wb = xw.Book()
sheet = wb.sheets[0]
添加数据
data = [['Month', '2019', '2020'],
        ['Jan', 50, 30],
        ['Feb', 40, 25],
        ['Mar', 30, 40],
        ['Apr', 25, 35]]
sheet.range('A1').value = data
创建图表
chart = sheet.charts.add()
chart.chart_type = 'line'
chart.set_source_data(sheet.range('A1:C5'))
chart.location = sheet.range('E5')
保存文件
wb.save('xlwings_chart.xlsx')
wb.close()
app.quit()