python - Groupby pandas 系列中具有相同键的连续元素

Question

我在 DataFrame 中有一个仅包含 0 或 1 的系列。我想获取每个 0 的“簇”的长度列表。这是一个玩具模型:

a=pd.DataFrame({'x':[1,1,0,0,1,0]})

我期望的是[2,1]

为此，我可以执行以下操作:

a[a['x']==0].index

然后使用 groupby 来查找连续索引，正如已经在 SO 上学到的那样。顺便问一下，有更有效的方法吗？

Answer 1

您可以使用 itertools.groupby 获得此结果.

from itertools import groupby

data = [1, 1, 0, 0, 1, 0]
grouper = (list(s) for _, s in groupby(data))
lengths = [len(s) for s in grouper if s[0] == 0]
print(lengths)

不确定pandas中的groupby是否编码相同。