'Count number of times value appears consecutively in a column

So I have this dataframe:

data = {'value':[1,1,1,0,1,0,1,0,0,0,0,0,1,1,1,0,0,1,0,1]}

df = pd.DataFrame(data) 
row value
0 1
1 1
2 1
3 0
4 1
5 0
6 1
7 0
8 0
9 0
10 0
11 0

I would like to add another column called "Cumulative" which will count how many times a number appears consecutively and will stop counting when the value is no longer the same as the one before. Then it should start counting again. This would be the resulting dataframe:

row value Cumulative
0 1 0
1 1 1
2 1 2
3 0 0
4 1 0
5 0 0
6 1 0
7 0 0
8 0 1
9 0 2
10 0 3
11 0 4
12 1 0

I have tried a couple of built-in functions like where, mask, and cumsum, but I'm honestly clueless when it comes to iterating and creating for loops and I'm positive that this is probably where the answer lies. Is there a function that I am not aware of that could do this? Or is there no avoiding for loops?



Solution 1:[1]

This code adds the column "Cumulative" and calculates the number of consecutive

# add col named "Cumulative"
df['Cumulative'] = [0 for i in range(len(df))]

last = 0
# if value is 1, add 1 to the value in the col named "Cumulative"
for i in range(len(df)):
    if df['value'][i] == 1:
        df['Cumulative'][i] = last + 1
        last += 1

    else:
        df['Cumulative'][i] = df['Cumulative'][i]
        last = 0
print(df)

The output is this:

    value  Cumulative
0       1           1
1       1           2
2       1           3
3       0           0
4       1           1
5       0           0
6       1           1
7       0           0
8       0           0
9       0           0
10      0           0
11      0           0
12      1           1
13      1           2
14      1           3
15      0           0
16      0           0
17      1           1
18      0           0
19      1           1

EDIT

This code now works for all different numbers.

import pandas as pd

data = {'value': [1, 1, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1, 0, 1]}

df = pd.DataFrame(data)

# add col named "Cumulative"
df['Cumulative'] = [0 for i in range(len(df))]

last_count = 0
# if value is 1, add 1 to the value in the col named "Cumulative"
last_num = df['value'][0]

for i in range(len(df)):

    if df['value'][i] == last_num:

        df['Cumulative'][i] = last_count + 1
        last_count += 1

    else:
        df['Cumulative'][i] = 1
        last_num = df['value'][i]
        last_count = 1

print(df)

The result:

    value  Cumulative
0       1           1
1       1           2
2       1           3
3       0           1
4       1           1
5       0           1
6       1           1
7       0           1
8       0           2
9       0           3
10      0           4
11      0           5
12      1           1
13      1           2
14      1           3
15      0           1
16      0           2
17      1           1
18      0           1
19      1           1

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source
Solution 1 marc_s