在工作簿中的新工作表上创建pandas数据透视表

时间:2016-07-22 16:50:29

标签: python pandas dataframe openpyxl

我正在尝试将我创建的数据透视表发送到工作簿中的新工作表中,但是,出于某种原因,当我执行代码时,使用数据透视表创建一个新工作表(工作表称为“工作表1”)并删除数据表。

这是我的代码:

worksheet2 = workbook.create_sheet()
worksheet2.title = 'Sheet1'
worksheet2 = workbook.active
workbook.save(filename)

excel = pd.ExcelFile(filename)
df = pd.read_excel(filename, usecols=['Product Description', 'Supervisor'])

table1 = df[['Product Description', 'Supervisor']].pivot_table(index='Supervisor', columns='Product Description', aggfunc=len, fill_value=0, margins=True, margins_name='Grand Total')



print table1

writer = pd.ExcelWriter(filename, engine='xlsxwriter')
table1.to_excel(writer, sheet_name='Sheet1')
workbook.save(filename)
writer.save()

另外,我的数据透视表设计遇到了一些麻烦。这是数据透视表的样子:

enter image description here

如何在总结每一行的末尾添加一列?像这样:(我只需要最后一栏,我不关心那样或任何东西的格式化)

enter image description here

1 个答案:

答案 0 :(得分:4)

调用pivot_table()

时,只需使用margins=Truemargins_name='Grand Total'参数即可

演示:

In [15]: df = pd.DataFrame(np.random.randint(0, 5, size=(10, 3)), columns=list('abc'))

In [16]: df
Out[16]:
   a  b  c
0  4  3  0
1  1  1  4
2  4  4  0
3  2  3  2
4  1  1  3
5  3  1  3
6  3  3  0
7  0  2  0
8  2  1  1
9  4  2  2

In [17]: df.pivot_table(index='a', columns='b', aggfunc='sum', fill_value=0, margins=True, margins_name='Grand Total')
Out[17]:
                c
b               1    2    3    4 Grand Total
a
0             0.0  0.0  0.0  0.0         0.0
1             7.0  0.0  0.0  0.0         7.0
2             1.0  0.0  2.0  0.0         3.0
3             3.0  0.0  0.0  0.0         3.0
4             0.0  2.0  0.0  0.0         2.0
Grand Total  11.0  2.0  2.0  0.0        15.0