將計數列添加到數據框，該數據框計算另一行何時更改

Question

我有一個數據框，其中包含如下列：

      x
0     1
1     1
2     0
3     1
4     0
5     0
6     0
7     1
8     1
9     1

我想添加一個每次x更改時都會計數的列，以便我的最終結果如下所示：

      x     y
0     1     0
1     1     0
2     0     1
3     1     2
4     0     3
5     0     3
6     0     3
7     1     4
8     1     4
9     1     4

如果沒有循環，我無法找到最快的方法。 如果y從0或1開始，我也不在乎。我確信我可以使用的是大熊貓的天賦。 你能幫我嗎？

PS。 我需要制作這個y列的原因是能夠按每個數字對行進行分組，如果有一種方法可以基本上完​​成同樣的事情而不創建它，那也可以。

Answer 1

diff你可以應用cumsum

df.x.diff().ne(0).cumsum()-1
Out[132]: 
0    0
1    0
2    1
3    2
4    3
5    3
6    3
7    4
8    4
9    4
Name: x, dtype: int32

Answer 2

使用Numpy陣列

注意 ：這也是object dtype的推廣，因為我們正在評估相等性。

df.assign(y=np.append(False, df.x.values[1:] != df.x.values[:-1]).cumsum())

   x  y
0  1  0
1  1  0
2  0  1
3  1  2
4  0  3
5  0  3
6  0  3
7  1  4
8  1  4
9  1  4