python - 如何找到多行配对(配对成员之间可能有行)

Question

抱歉，我发现问题标题有些含糊，我不确定如何表述问题。希望这个例子能让我清楚地了解我想要做的事情。

我有一个像这样的数据框:

    Time    Event
0    .1      start
1    .2      end
2    .3      start
3    .4      foo
4    .5      bar
5    .6      end

我正在尝试获取相应开始行和结束行的索引(在本例中，是 (0,1) 和 (2,5) 对)。

我不完全确定如何编写此代码的矢量化版本。我可以很容易地在一个循环中完成它，但我正在努力避免它。

可以做出的假设:

-每一个开始都有一个关联的结束

-开始和结束之间可以有其他事件

如有任何帮助，我们将不胜感激。

编辑:我意识到我的问题遗漏了一个关键案例。我可以让一个事件在另一个事件中开始。所以像这样的数据框:

    Time    Event
0    .1      start
1    .2      start
2    .3      end
3    .4      foo
4    .5      bar
5    .6      end

结果应该是 (0,2), (1,5)

我意识到这完全改变了问题的性质。很抱歉没有早点提到它。堆栈是处理此问题的最简单方法吗？

Answer 1

获取两个不同列表中的索引并连接列表以创建元组

start = df[(df['Event'] == 'start')].index
end = df[(df['Event'] == 'end')].index
[(i1, i2) for i1, i2 in zip(start, end) if i1 < i2]

你得到

[(0, 1), (2, 5)]

它将处理任何非偶数对的开始结束。使用更新后的数据框，

    Time    Event
0    .1      start
1    .2      start
2    .3      end
3    .4      foo
4    .5      bar
5    .6      end

这个解决方案给出了

[(0, 2), (1, 5)]