在使用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 函数的使用及参数的调参请参见官方函数说明。