Увеличивать столбец, если строка появляется в кадре данных

Question

Я хочу увеличить столбец s_number, если слово «LIVE» появляется в другом столбце. Первый столбец всегда будет «LIVE», а затем любое вхождение «LIVE» всегда будет в группе 2. Я хочу увеличить столбец s_number после второго вхождения, я не уверен, как это сделать

 Electric  s_number
   LIVE      1
   EARTH     1
   NEUTRAL   1
   NEUTRAL   1
   LIVE      1
   LIVE      2
   EARTH     2
   NEUTRAL   2

Answer 1

Используя shift и сам column для создания bool, затем выполните cumsum

((df.Electric=='LIVE')&(df.Electric.shift()=='LIVE')).cumsum()+1
Out[278]: 
0    1
1    1
2    1
3    1
4    1
5    2
6    2
7    2
Name: Electric, dtype: int32