python 创建一个excel,写入5行3列值并且在一个新的sheet页生成对应的数据透视图
时间: 2024-03-24 07:37:48 浏览: 117
好的,可以使用 Python 中的 openpyxl 库来创建 Excel 文件,并使用 pandas 库来生成数据透视表。以下是示例代码:
```python
import openpyxl
import pandas as pd
# 创建 Excel 文件并添加一个名为 data 的 sheet 页
wb = openpyxl.Workbook()
sheet = wb.active
sheet.title = 'data'
# 写入数据
data = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12],
[13, 14, 15]
]
for row in data:
sheet.append(row)
# 创建一个名为 pivot 的 sheet 页,并生成数据透视表
pivot_sheet = wb.create_sheet('pivot')
df = pd.DataFrame(data, columns=['A', 'B', 'C'])
pivot_table = pd.pivot_table(df, values=['A', 'B', 'C'], aggfunc='sum')
for i, row in enumerate(pivot_table.values):
pivot_sheet.append([pivot_table.index[i][0], row[0], row[1], row[2]])
# 保存 Excel 文件
wb.save('example.xlsx')
```
这段代码将会创建一个名为 `example.xlsx` 的 Excel 文件,并在其中添加两个 sheet 页:`data` 和 `pivot`。`data` sheet 页中包含了 5 行 3 列的数据,而 `pivot` sheet 页中则是生成的数据透视表。注意,这里我们使用了 pandas 库来生成数据透视表,并将其写入到 Excel 文件中。
阅读全文