分组并计数

对一个大文件中的某列进行分组并对其他列计数和求和

import pandas as pd
import numpy as np
import os

data =pd.read_csv(r'C:\Users\lenovo\Desktop\12\total.csv').drop_duplicates(['快递员电话'])#去重
data_1 =pd.read_csv(r'C:\Users\lenovo\Desktop\12\total.csv')

data1 = data.groupby('快递品牌').aggregate({'快递员姓名':'count'}) #分组并分别对其他列做数(去重对快递员求和）
data1 = pd.concat([data1,data_1.groupby('快递品牌').aggregate({'投件量':'sum'})],axis =1).sort_values(by='投件量')

if os.path.exists(r'C:\Users\lenovo\Desktop\12\total1.csv'): #判断文件是否存在
    os.remove(r'C:\Users\lenovo\Desktop\12\total1.csv')      #如果存在文件则删除
data1.to_csv(r'C:\Users\lenovo\Desktop\12\total1.csv')