將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