【python 数据处理】分组求和、合并

# encoding: utf-8
import sys
reload(sys)
sys.setdefaultencoding('utf-8')


import time
import pandas as pd
import xlsxwriter


time1 = time.time()
import numpy as np
#############读取数据##########################
df=pd.read_excel('C:/pic/sample.xlsx')

# 建立每个顾客的购买记录


# print df

##########################什么时候购买，买的产品单价###################################
data1=pd.DataFrame({"Customer Account":df.iloc[:,4],"Date":df.iloc[:,0],"Product Name":df.iloc[:,3],"Qty":df.iloc[:,5],"Sale Price":df.iloc[:,7]})

####################买了多少产品##################################
df2 = data1.groupby(by=['Customer Account'])['Qty'].sum()

data2=pd.DataFrame({"Customer Account":df2.index,"count":df2})

###################合并data1和data2##############################

data3=pd.merge(data1,data2,how='left',on='Customer Account')

###########写出excel#################

pd.DataFrame.to_excel(data3,'c:/result.xlsx',index=False)
print u'数据处理完成'