Python openpyxl库:用代码创建Excel数据透视表
使用openpyxl库可以打开一个有数据的excel文件,并且可以使用openpyxl库进行数据透视表的操作。下面是一个示例代码:
from openpyxl import Workbook
from openpyxl import load_workbook
from openpyxl.utils.dataframe import dataframe_to_rows
from openpyxl.pivot import PivotTable, PivotTableStyleInfo
# 打开excel文件
filename = 'data.xlsx'
wb = load_workbook(filename)
# 选择工作表
ws = wb.active
# 将工作表数据转换为DataFrame
data = ws.values
cols = next(data)
df = pd.DataFrame(data, columns=cols)
# 创建一个新的工作簿和工作表
new_wb = Workbook()
new_ws = new_wb.active
# 将DataFrame数据写入新的工作表
for row in dataframe_to_rows(df, index=False, header=True):
new_ws.append(row)
# 创建数据透视表
pivot_table = PivotTable(xWorksheet=new_ws, sourceReference=new_ws.dimensions, targetCell='A1', name='PivotTable')
# 设置数据透视表样式
style = PivotTableStyleInfo(name='PivotStyleLight16')
pivot_table.pivotTableStyleInfo = style
# 保存新的工作簿
new_wb.save('pivot_table.xlsx')
在上面的代码中,首先使用load_workbook函数打开一个带有数据的excel文件,并选择工作表。然后,将工作表中的数据转换为DataFrame格式。接下来,创建一个新的工作簿和工作表,并将DataFrame数据写入新的工作表。然后,通过PivotTable类创建数据透视表,并设置数据透视表的样式。最后,将新的工作簿保存为一个新的excel文件。
原文地址: https://www.cveoy.top/t/topic/pUDe 著作权归作者所有。请勿转载和采集!