python - 每次下一行与前一行不同时增加一个值

Question

我有这个 df :

name
str_a
str_a
str_a
str_b
str_b
str_c
str_c

在专栏 name我得到了不同的 str ，我想创建另一个名为 number 的列它的第一行是 0，然后每次前一个名称与下一个名称不同时，它都会在前一个数字上加 +1。预期的结果是:

name  number
str_a 0
str_a 0
str_a 0
str_b 1
str_b 1 
str_c 2 
str_c 2

有没有办法在没有循环的情况下做到这一点？
谢谢 !

Answer 1

让我们试试 shift与 cumsum , 通知您可以查看df['name'].factorize()[0] ，但是，如果名称重复但并不总是连续连续factorize不管用

df['name'].ne(df['name'].shift()).cumsum()-1
Out[10]: 
0    0
1    0
2    0
3    1
4    1
5    2
6    2
Name: name, dtype: int32