python 中将数据生成为Excel文件,并且汇总图表或者表格
时间: 2024-04-15 22:31:28 浏览: 116
使用Python把数据写入Excel
3星 · 编辑精心推荐
要将数据生成为Excel文件并创建汇总图表或表格,你可以使用Python中的第三方库,如`pandas`和`openpyxl`。下面是一个示例代码:
```python
import pandas as pd
from openpyxl import Workbook
from openpyxl.chart import BarChart, Reference
# 创建数据
data = {
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'Salary': [5000, 6000, 7000, 8000]
}
# 将数据转换为DataFrame
df = pd.DataFrame(data)
# 创建Excel文件并将数据写入其中
wb = Workbook()
ws = wb.active
ws.title = 'Data'
for r in dataframe_to_rows(df, index=False, header=True):
ws.append(r)
# 创建汇总表格
summary_data = {
'Average Age': [df['Age'].mean()],
'Average Salary': [df['Salary'].mean()]
}
summary_df = pd.DataFrame(summary_data)
summary_ws = wb.create_sheet(title='Summary')
for r in dataframe_to_rows(summary_df, index=False, header=True):
summary_ws.append(r)
# 创建柱状图
chart = BarChart()
values = Reference(summary_ws, min_col=2, max_col=3, min_row=2, max_row=2)
categories = Reference(summary_ws, min_col=1, max_col=1, min_row=3, max_row=3)
chart.add_data(values)
chart.set_categories(categories)
summary_ws.add_chart(chart, 'E1')
# 保存Excel文件
wb.save('data.xlsx')
```
在示例代码中,首先创建了一个包含姓名、年龄和薪水数据的字典。然后将数据转换为`pandas`的`DataFrame`对象。接下来,使用`openpyxl`创建一个新的Excel工作簿,并将数据写入名为"Data"的工作表中。
然后,创建一个名为"Summary"的新工作表,并在其中计算平均年龄和平均薪水,并将结果写入汇总表格。随后,创建一个柱状图并将其添加到汇总工作表中。
最后,使用`wb.save()`将Excel文件保存到指定路径(此处为"data.xlsx")。
这只是一个示例,你可以根据实际需求进行修改和扩展。你可以通过`pandas`和`openpyxl`库提供的各种功能来创建更复杂的表格和图表。
阅读全文