I want to program a counter with python which counts the "1-values" in the column "values". The counter has to start counting after the first 0 value in the rows.
As it is seen in the example:
This is just an example. In reality, the table has more than three "1-values" at the beginning. Therefore it's desired that the code should be automated.
| Values | Resulted counter values |
|---|---|
| 1 | 0 |
| 1 | 0 |
| 1 | 0 |
| 0 | 0 |
| 1 | 1 |
| 1 | 2 |
| 1 | 3 |
| 1 | 4 |
| 0 | 0 |
| 1 | 1 |
| 1 | 2 |
| 1 | 3 |
You can use groupby.cumcount and a mask with where:
# is the value not a 1?
m = df['Values'].ne(1)
# for each group starting with "not-a-1", cumcount,
# mask the values before the first non-1
df['counter'] = df.groupby(m.cumsum()).cumcount().where(m.cummax(), 0)
output:
Values counter
0 1 0
1 1 0
2 1 0
3 0 0
4 1 1
5 1 2
6 1 3
7 1 4
8 0 0
9 1 1
10 1 2
11 1 3
Intermediates:
Values counter m m.cumsum() cumcount m.cummax()
0 1 0 False 0 0 False
1 1 0 False 0 1 False
2 1 0 False 0 2 False
3 0 0 True 1 0 True
4 1 1 False 1 1 True
5 1 2 False 1 2 True
6 1 3 False 1 3 True
7 1 4 False 1 4 True
8 0 0 True 2 0 True
9 1 1 False 2 1 True
10 1 2 False 2 2 True
11 1 3 False 2 3 True