dataframe：将一列中的值与上一列中的值进行比较

Question

我有一个数据框，我想在其中添加一列（col_new），该列的值取决于现有列（col_exist）中的值的比较

现有列（类型=对象）包含As和Bs。 新列应从1开始计数。 如果A跟在A后面，则计数应增加1。 如果A跟在B后面，则计数应增加1。 如果a之后是B，则计数不应上升。 如果a B跟随a B，则计数不应上升

col_exist  col_new
    A       1
    A       2
    A       3
    B       3
    A       4
    B       4
    B       4
    A       5
    B       5

我对编程一无所知，所以提前感谢您的adequade回答

Answer 1

使用eq和cumsum：

df['col_new'] = df['col_exist'].eq('A').cumsum()

输出：

  col_exist  col_new
0         A        1
1         A        2
2         A        3
3         B        3
4         A        4
5         B        4
6         B        4
7         A        5
8         B        5