Pandas对数据进行排序标记（rank函数）

现有一个问题：
数据是多个重复两遍的user_id，想新建一列，给重复的user_id标号，1和2。
具体功能如图：

使用groupby和rank函数即可实现上述功能。
具体代码如下：

# 注： 这里是用user_id列对数据进行分组，然后对id列进行排序并且标号
df['rank_id'] = df.groupby('user_id').id.rank(ascending=