Как исправить функцию np.cumsum при отображении ряда с регулярным выражением

Question

У меня есть список вроде:

**x** 
Chapter 1 
some text
Chapter 2
Chapter 3
Chapter 4
Chapter 5"

Вывод, который я ищу, - это серия вроде:

1
1
2
3
4
5

Я написал этот код

pattern = r"chapter \d" 
x.map(lambda x: np.cumsum(bool(re.search(pattern,str(x), flags=re.I))))

Это дает мне вывод:

[1]
[0]
[1]
[1]
[1]
[1]

Можете ли вы, ребята, помочь мне исправить этот код? Вы также можете предложить лучшее решение для этого. Спасибо

Answer 1

Спасибо, ребята, за помощь в исправлении моего кода. Я видел альтернативное решение с использованием ffill, и это было здорово.

Я использовал решение с 2 вкладышами ниже:

pattern = r"chapter \d" 
y = x.map(lambda x: bool(re.search(pattern,str(x), flags=re.I))) 
z = np.cumsum(y)