python 生成数据透视表后写入 Excel

如何使用 Python 对 Excel 做一份数据透视表

如上贴中,得到数据透视表后,我们如何写入 Excel 呢?
如果我们将得到的返回值直接写入 Excel,会发现,会出现丢失第一列(索引列)的数据。

解决方案:
方法 1、将返回值转换成字符串, 代码如下:

# res 为上一篇帖子的返回值。
a = str(res).split('\n')
newlist = []
for i in a:
	i.split()
	newlist.append(i.split())
n_list = newlist[3:]
n_list.insert(0,['商户编号','交易金额', '入账金额', '商户手续费'])
iexcel.write_cell(path='C:/iS-RPA/studio-v6/工作簿1.xlsx',sheet=1,text=n_list)

缺点,需要对不同的项目,作不同的调整。

方法 2、代码如下:

import  pandas  as pd
import  numpy  as np

df =  pd.read_excel(io='C:/file/test/test1/1904农行.xlsx', sheet_name=2)
res=pd.pivot_table(df,index=['商户编号'],aggfunc=[np.sum])

result_sum=pd.DataFrame(res.sum()).T
result_pivot_sum=res.append(result_sum)
result_pivot_sum=result_pivot_sum.rename(index={0:u'总计'})
result_pivot_sum.to_excel('//mac/Home/Desktop/data.xlsx',encoding="utf-8")

如上代码,便可轻松将生成的数据透视表的数据完整的、一次性的写入 Excel 了。