在pandas数据帧中查找连续的片段

9 浏览
0 Comments

在pandas数据帧中查找连续的片段

我有一个pandas.DataFrame,其中记录了连续时间点上的测量结果。每个测量结果相应地对应了系统在每个时间点的不同状态。因此,DataFrame中还包含了系统在每个测量点上的状态列。状态的变化比测量间隔要慢得多。因此,表示状态的列可能如下所示(索引:状态):

1:  3
2:  3
3:  3
4:  3
5:  4
6:  4
7:  4
8:  4
9:  1
10: 1
11: 1
12: 1
13: 1

是否有一种简单的方法可以获取连续相等状态的每个片段的索引?也就是说,我希望得到类似于以下的结果:

[[1,2,3,4], [5,6,7,8], [9,10,11,12,13]]

结果也可以是其他形式,而不仅仅是普通列表。

到目前为止,我想到的唯一解决办法是手动迭代每一行,找到状态变化点,并从这些变化点重构索引。但我希望能有一个更简单的解决办法。

0