python - 如何在 Pandas : count, 删除列、删除重复项中更好地执行此步骤

Question

这是我的新日常事务，我处理带有许多列的df，其中包括这两列:user 和事件。我计算每个用户的事件数量，向原始df添加一个新列count。然后，我只保留存在多个相同行的 user 和 count，然后执行 drop_duplicates() 删除重复项，从而获取事件 每个用户的计数。我确信我正在做一些多余的工作。

完成此类任务的优雅方式是什么。

df['count'] = df.groupby('user')['event'].transform('count')
df = df[['user','count']]
df = df.drop_duplicates()
plt.bar(x=df['user'], height=df['count'])

Answer 1

使用GroupBy.count对于 Series，然后调用 Series.plot.bar :

df.groupby('user')['event'].count().plot.bar()