基於2個分類列pandas數據框創建新的增量列

Question

我有一個熊貓數據框，其中包含username和phase列。 我想用增量值創建一個名為count的單獨列。

count將基於username在特定phase出現的次數。 我怎樣才能有效地做到這一點？ 任何建議表示贊賞。

    username  phase      count
0    andrew    1          1
1    andrew    1          2
2    alex      1          1
3    alex      2          1
4    andrew    1          3
5    cindy     3          1
6    alex      2          2

Answer 1

您可以在username和phase的 groupby 之后使用cumcount 。

df['count'] = df.groupby(['username', 'phase']).cumcount()+1
print(df)

---

  username  phase  count
0   andrew      1      1
1   andrew      1      2
2     alex      1      1
3     alex      2      1
4   andrew      1      3
5    cindy      3      1
6     alex      2      2