使用取决于另一列的整数序列将列添加到数据框答案

【问题标题】：add column to dataframe with sequence of integers depending on another column使用取决于另一列的整数序列将列添加到数据框
【发布时间】：2022-07-11 23:54:35
【问题描述】：

df =  pd.DataFrame({'A':[3,5,2,5,4,2,5,2,3,1,4,1], 'B':['x','y','x','x','y','z','z','x','y','y','x','z']})

我想添加一列 C，对于 B 中的每个字母，它都包含连续整数：

    A  B  C
0   3  x  1
1   5  y  1
2   2  x  2
3   5  x  3
4   4  y  2
5   2  z  1
6   5  z  2
7   2  x  4
8   3  y  3
9   1  y  4
10  4  x  5
11  1  z  3

【问题讨论】：

必须是 dup：df['C'] = df.groupby('B').cumcount() + 1
这能回答你的问题吗？ How to add sequential counter column on groups using Pandas groupby

标签： python pandas

【解决方案1】：

你可以使用cumcount()按B分组

df =  pd.DataFrame({'A':[3,5,2,5,4,2,5,2,3,1,4,1], 'B':['x','y','x','x','y','z','z','x','y','y','x','z']})
df['C'] = df.groupby('B').cumcount() + 1

【讨论】：