python - 删除可能不必要的代码行

我目前正在将第三列添加到dataFrame中：

per_day           =    dg.groupby(dg['DateTime'].dt.date)['Burned'].count().reset_index(name='Trx')
per_day_burned    =    dg.groupby(dg['DateTime'].dt.date)['Burned'].sum().reset_index(name='Burned')
per_day['Burned'] = per_day_burned['Burned']

在这里，我将per_day ['Burned']添加到per_day。

有没有一种方法可以在没有第二行代码的情况下直接将其添加到per_day dataFrame中？

最佳答案

您可以通过agg在同一组上使用多个聚合。

dg.groupby(dg['DateTime'].dt.date)['Burned'].agg(['count', 'sum'])