python - 如何为数据框中的列应用多个条件？

Question

我有一个如下所示的数据框，两列之间没有关系


物品
第 1 列


猫
1

猫
1

狗
1

鱼
1

鱼
1

鱼
2

蛇
2

蛇
2

蛇
2


我想使用这两个生成一个新列。结果列应该对项目列进行累积总和(仅在项目更改时才增加编号)，并且如果 col1 中的系列发生更改，则编号必须再次从 1 开始。


物品
第 1 列
结果


猫
1
1

猫
1
1

狗
1
2

鱼
1
3

鱼
1
3

鱼
2
1

蛇
2
2

蛇
2
2

蛇
2
2


我能够用下面的代码做累积总和，

df["result"] = (df["item"] != df["item"].shift(1)).cumsum()

但是当 col1 发生变化时将计数器重置为 1 的条件是我需要的。

Answer 1

查看 groupby和 factorize内transform

df['new'] = df.groupby('col1').item.transform(lambda x : x.factorize()[0]+1)
df
    item  col1  new
0    cat     1    1
1    cat     1    1
2    dog     1    2
3   fish     1    3
4   fish     1    3
5   fish     2    1
6  snake     2    2
7  snake     2    2
8  snake     2    2