在pandas数据帧中创建一个count列

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