python - 计算标记的特定 block

Question

我想根据 LabelID 属性计算数据帧中“1” block 的数量。例如，给定以下数据框:

DF 输入:

   eventTime                 velocity     LabelId
1  2017-08-19 12:53:55.050         3        0
2  2017-08-19 12:53:55.100         4        1
3  2017-08-19 12:53:55.150       180        1
4  2017-08-19 12:53:55.200         2        1
5  2017-08-19 12:53:55.250         5        0
6  2017-08-19 12:53:55.050         3        0
7  2017-08-19 12:53:55.100         4        1
8  2017-08-19 12:53:55.150        70        1
9  2017-08-19 12:53:55.200         2        1
10 2017-08-19 12:53:55.250         5        0

Output=2，因为它有两个 1 block 。Block_1=rows 2-4 和 Block_2=rows 7-9。请提供任何帮助，我们将不胜感激。

最诚挚的问候，卡洛

Answer 1

我们可以使用diff() 。像这样的事情:

d = df.LabelId.diff()
d.iloc[0] = df.LabelId.iloc[0]

这给你:

[0, 1, 0, 0, -1, 0, 1, 0, 0, -1]

1的组数就是diff为1的次数。所以:

(d == 1).sum()

给你答案。