如何在一个 groupby 列上执行聚合选项，给出两列输出

Question

我正在对 groupby 数据框执行一堆聚合统计信息。特别是对于一列ios_id，我想要一个计数和一个不同的计数。我不确定如何将其输出到具有不同名称的两个独立列。截至目前，不同的计数只是覆盖计数。

如何将 ios_id 列的不同计数和计数输出到两个单独的列？

df_new = df.groupby('video_id').agg({"ios_id": np.count_nonzero,
                                     "ios_id": pd.Series.nunique,
                                     "feed_position": np.average,
                                     "time_watched": np.sum,
                                     "video_length": np.sum}).sort('ios_id', ascending=False)

Answer 1

这样的事情应该可以工作。注意 iOS_id 的嵌套字典结构。

df_new = df.groupby('video_id').agg({"ios_id": {"count": "count",
                                                "distinct": "unique"},
                                     "feed_position": np.average,
                                     "time_watched": np.sum,
                                     "video_length": np.sum})

更多详情请参考Naming returned columns in Pandas aggregate function：