python - 我怎样才能得到一个组的位置？

Question

我有这样的 Pandas 数据框

data = [[1, 'a'], [2, 'a'], [3, 'b'], [4, 'b'], [5, 'a'], [6, 'c']]
df1 = pd.DataFrame(data, columns=['Id', 'Group'])

Id    Group
 1      a
 2      a
 3      b
 4      b
 5      a
 6      c

在不改变顺序的情况下，我需要根据“组”获取每个 Id 的位置。

基本上，我想要低于输出

Id    Group   position
 1      a         1
 2      a         2
 3      b         1
 4      b         2
 5      a         3
 6      c         1

Answer 1

尝试，转换 + 累积计数

df1['position'] = df1.groupby('Group').transform('cumcount') + 1

---

   Id Group  position
0   1     a         1
1   2     a         2
2   3     b         1
3   4     b         2
4   5     a         3
5   6     c         1