将COUNTIFS列添加到Pandas DataFrame

Question

我正在使用Python中的Pandas DataFrame，它包含下图的前两列：user_id，user_time。

我只需要向其添加第三列，这基本上是每个“ user_id”的“ user_time”序列。 在Excel中，公式为：

=COUNTIFS([user_id],[@[user_id]], [user_time], "<="&[@[user_time]])

请问如何在Python中执行此操作？

Answer 1

你需要cumcount 。

感谢Ding的评论-如有必要，请使用sort_values ：

df = df.sort_values(['user_id', 'user_time'])
df['COUNTIFS'] = df.groupby('user_id').cumcount() + 1