我想仔细地保存一个熊猫数据透视表,并将其格式化为Excel工作簿。
我有一个基于此公式的熊猫数据透视表:
table = pd.pivot_table(d2, values=['Wert'], index=['area', 'Name'], columns=['Monat'],
aggfunc={'Wert': np.sum}, margins=True).fillna('')
根据我的原始数据框:
df
Area Name2 Monat Wert
A A 1 2
A A 1 3
A B 1 2
A A 2 1
所以数据透视表如下所示:
Wert
Monat 1 2 All
Area Name
A A 5 1 6
B 2 2
All 7 1 8
然后,我想使用以下代码将其保存在excel工作簿中:
import pandas as pd
import xlsxwriter
workbook = xlsxwriter.Workbook('myexcel.xlsx)
worksheet1 = workbook.add_worksheet('table1')
caption = 'Table1'
worksheet1.set_column(1, 14, 25) #irrelevant, just a random size right now
worksheet1.write('B1', caption)
worksheet1.add_table('B3:K100', {'data': table.values.tolist()}) #also wrong size from B3 to K100
workbook.close()
如何调整它并在Excel中保存数据透视表?
如果我使用的是熊猫命令.to_excel
,则如下所示:
哪个很好,但是列名不考虑名称的宽度,并且背景色看起来也不好,而且我也缺少捕获。
保存前使用数据框reset_index()函数。
Pandas to_csv函数不适合保存美观的excel文件。
我找到了此topic:的组合的解决方案
flattened = pd.DataFrame(table.to_records())
flattened.columns = [column.replace("('Wert', ", "Monat: ").replace(")", "") for column in flattened.columns] ##only for renaming the column headers
然后:
workbook = xlsxwriter.Workbook(excelfilename, options={'nan_inf_to_errors': True})
worksheet = workbook.add_worksheet('Table1')
worksheet1.set_column(0, flattned.shape[1], 25)
worksheet.add_table(0, 0, flattened.shape[0], flattened.shape[1]-1,
{'data': flattened.values.tolist(),
'columns': [{'header': c} for c in flattened.columns.tolist()],
'style': 'Table Style Medium 9'})
workbook.close()