如何对Dataframe列中当前连续的0进行计数？

import pandas as pd

# 创建一个示例Dataframe
df = pd.DataFrame({'column': [0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0]})

# 使用shift函数创建一个新的列，表示当前元素与前一个元素是否相等
df['is_zero'] = (df['column'] == df['column'].shift())

# 使用cumsum函数创建一个新的列，表示当前元素与前一个元素是否相等的累积和
df['cumulative_sum'] = df['is_zero'].cumsum()

# 使用groupby函数和transform函数计算每个连续0序列的长度
df['consecutive_zeros'] = df.groupby('cumulative_sum').transform('size')

# 过滤出值为0的行，并选择consecutive_zeros列作为结果
result = df[df['column'] == 0]['consecutive_zeros']

# 打印结果
print(result)

0    3
1    3
2    3
5    4
6    4
7    4
10   2
11   2
Name: consecutive_zeros, dtype: int64