pandas中的groupby函数的分组结果怎么保存成DataFrame

在使用pandas进行数据统计分析时,作为一个小白纠结了好久如何保存groupby函数的分组结果,本打算放弃了,一个偶然的机会看前人分享的代码才发现了可以通过reset_index()函数将groupby()的分组结果转换成DataFrame对象。

代码举例:

total = df.groupby(['al_m6_id_notbank_allnum'])['target_cpd'].count()
total = pd.DataFrame({'total':total})
bad = df.groupby(['al_m6_id_notbank_allnum'])['target_cpd'].sum()
bad = pd.DataFrame({'bad':bad})
regroup =  total.merge(bad,left_index=True,right_index=True, how='left')
regroup.reset_index(inplace=True)

关于 reset_index 函数的使用及参数的调参请参见官方函数说明。

你可能感兴趣的:(python)