在pandas數據框中創建計數列

Question

如何快速添加一列來計算[user，class]組的重復次數。

例如;

user    class   val
1   2   23
1   2   23
1   3   29
1   3   29
1   4   11

代碼對user和class進行分組，並創建一列，范圍從1（或0）到該組的出現次數。

user    class   Mean    n_count
1   2   23  1
1   2   23  2
1   3   29  1
1   3   29  2
1   4   11  1

Answer 1

您可以使用groupby然后使用cumcount ：

>>> df["n_count"] = df.groupby(["user", "class"]).cumcount() + 1
>>> df
   user  class  val  n_count
0     1      2   23        1
1     1      2   23        2
2     1      3   29        1
3     1      3   29        2
4     1      4   11        1