python - 如何计算列表中值 block 的大小？

Question

我有一个这样的 list :

list_1 = [0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1]

如何计算此列表中 1和 0值的块大小？
结果列表如下所示:

list_2 = [1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 1, 1]

Answer 1

尝试使用cumsum和diff，然后使用transform count

s = pd.Series(list_1)
s.groupby(s.diff().ne(0).cumsum()).transform('count')
Out[91]: 
0     1
1     2
2     2
3     3
4     3
5     3
6     4
7     4
8     4
9     4
10    1
11    1
dtype: int64