如何用Python写Excel图表

要用Python写Excel图表，可以使用诸如openpyxl、pandas和XlsxWriter等库。首先，安装所需的库、创建一个工作簿、添加数据、生成图表是基本步骤。下面，我将详细介绍如何用Python写Excel图表。

一、安装所需库

首先，我们需要安装openpyxl和XlsxWriter库。这些库可以通过pip安装：

pip install openpyxl XlsxWriter pandas

二、创建工作簿并添加数据

要创建一个Excel工作簿并添加数据，我们可以使用pandas库。以下是一个简单的示例：

import pandas as pd
创建示例数据
data = {
    'Month': ['January', 'February', 'March', 'April', 'May', 'June'],
    'Sales': [150, 200, 300, 400, 500, 600]
}
创建DataFrame
df = pd.DataFrame(data)
将DataFrame保存到Excel文件
df.to_excel('sales_data.xlsx', index=False)

三、使用XlsxWriter生成图表

接下来，我们使用XlsxWriter库来生成图表：

import pandas as pd
import xlsxwriter
读取数据
df = pd.read_excel('sales_data.xlsx')
创建一个XlsxWriter工作簿对象
workbook = xlsxwriter.Workbook('sales_chart.xlsx')
添加一个工作表
worksheet = workbook.add_worksheet()
将DataFrame写入工作表
for row_num, (index, row) in enumerate(df.iterrows()):
    worksheet.write_row(row_num + 1, 0, row)
创建一个图表对象
chart = workbook.add_chart({'type': 'line'})
配置图表数据系列
chart.add_series({
    'categories': ['Sheet1', 1, 0, len(df), 0],  # 'Sheet1'指工作表名称
    'values': ['Sheet1', 1, 1, len(df), 1],
    'name': 'Sales Data'
})
将图表插入工作表
worksheet.insert_chart('D2', chart)
关闭工作簿
workbook.close()

解释：

创建一个XlsxWriter工作簿对象：我们首先创建一个XlsxWriter工作簿对象，并添加一个工作表。
将DataFrame写入工作表：我们使用write_row方法将DataFrame的数据写入工作表中。
创建一个图表对象：我们创建一个图表对象，并配置图表数据系列。
将图表插入工作表：我们将图表插入到工作表中的指定位置。
关闭工作簿：最后，我们关闭工作簿以保存更改。

四、定制图表

除了基本的图表，我们还可以定制图表的外观和格式。以下是一些常见的定制选项：

1、设置图表标题和轴标签

我们可以为图表添加标题和轴标签：

chart.set_title({'name': 'Monthly Sales Data'})
chart.set_x_axis({'name': 'Month'})
chart.set_y_axis({'name': 'Sales'})

2、设置图表样式

我们可以设置图表的样式，例如线条颜色和宽度：

chart.add_series({
    'categories': ['Sheet1', 1, 0, len(df), 0],
    'values': ['Sheet1', 1, 1, len(df), 1],
    'name': 'Sales Data',
    'line': {'color': 'blue', 'width': 2.25}
})

3、添加数据标签

我们可以为图表添加数据标签，以显示每个数据点的值：

chart.add_series({
    'categories': ['Sheet1', 1, 0, len(df), 0],
    'values': ['Sheet1', 1, 1, len(df), 1],
    'name': 'Sales Data',
    'data_labels': {'value': True}
})

五、使用OpenPyXL生成图表

除了使用XlsxWriter，我们还可以使用openpyxl库生成图表。以下是一个示例：

import openpyxl
from openpyxl.chart import LineChart, Reference
加载数据
workbook = openpyxl.load_workbook('sales_data.xlsx')
sheet = workbook.active
创建一个图表对象
chart = LineChart()
添加数据
data = Reference(sheet, min_col=2, min_row=1, max_col=2, max_row=7)
categories = Reference(sheet, min_col=1, min_row=2, max_row=7)
chart.add_data(data, titles_from_data=True)
chart.set_categories(categories)
设置图表标题和轴标签
chart.title = "Monthly Sales Data"
chart.x_axis.title = "Month"
chart.y_axis.title = "Sales"
将图表插入到工作表
sheet.add_chart(chart, "D2")
保存工作簿
workbook.save('sales_chart_openpyxl.xlsx')

解释：

加载数据：我们使用openpyxl加载已有的Excel文件。
创建一个图表对象：我们创建一个LineChart对象。
添加数据：我们使用Reference对象来引用数据，并添加到图表中。
设置图表标题和轴标签：我们为图表设置标题和轴标签。
将图表插入到工作表：我们使用add_chart方法将图表插入到工作表中的指定位置。
保存工作簿：最后，我们保存工作簿以保存更改。

六、使用pandas生成图表

pandas库也提供了生成图表的功能。以下是一个使用pandas生成图表的示例：

import pandas as pd
创建示例数据
data = {
    'Month': ['January', 'February', 'March', 'April', 'May', 'June'],
    'Sales': [150, 200, 300, 400, 500, 600]
}
创建DataFrame
df = pd.DataFrame(data)
生成图表
ax = df.plot(x='Month', y='Sales', kind='line', title='Monthly Sales Data')
保存图表到文件
fig = ax.get_figure()
fig.savefig('sales_chart_pandas.png')

解释：

创建示例数据：我们创建一个包含月份和销售数据的DataFrame。
生成图表：我们使用plot方法生成一个线图，并设置图表标题。
保存图表到文件：我们使用get_figure方法获取图表对象，并将其保存为PNG文件。

七、综合使用多个库

在实际应用中，我们可以综合使用多个库，以获得更强大的功能。例如，我们可以使用pandas进行数据处理，openpyxl进行Excel文件操作，XlsxWriter进行图表生成：

import pandas as pd
import openpyxl
from openpyxl.chart import LineChart, Reference
import xlsxwriter
创建示例数据
data = {
    'Month': ['January', 'February', 'March', 'April', 'May', 'June'],
    'Sales': [150, 200, 300, 400, 500, 600]
}
创建DataFrame
df = pd.DataFrame(data)
使用pandas保存数据到Excel文件
df.to_excel('sales_data_combined.xlsx', index=False)
使用openpyxl加载Excel文件
workbook = openpyxl.load_workbook('sales_data_combined.xlsx')
sheet = workbook.active
使用XlsxWriter创建一个图表对象
chart = LineChart()
添加数据
data = Reference(sheet, min_col=2, min_row=1, max_col=2, max_row=7)
categories = Reference(sheet, min_col=1, min_row=2, max_row=7)
chart.add_data(data, titles_from_data=True)
chart.set_categories(categories)
设置图表标题和轴标签
chart.title = "Monthly Sales Data"
chart.x_axis.title = "Month"
chart.y_axis.title = "Sales"
将图表插入到工作表
sheet.add_chart(chart, "D2")
使用openpyxl保存工作簿
workbook.save('sales_chart_combined.xlsx')