按索引计算pandas数据帧上的连续事件

Question

我尝试使用数据框来选择事件。让我们说我只有两种事件：1，-1和以下数据帧：

a=pd.DataFrame({'ev':[1,1,-1,1,1,1,-1,-1]})

使用：

a[a['ev']>0]

我只能选择1个事件。现在我想算一下我连续多少次活动;在这种情况下，我想得到[2,3]

关键是：如何获取a[a['ev']>0]中涉及的索引的列表（或系列）？

Answer 1

添加.index以返回索引列表

In [43]: a=pd.DataFrame({'ev':[1,1,-1,1,1,1,-1,-1]})

In [44]: b = list(a[a['ev']>0].index)

In [45]: b
Out[45]: [0, 1, 3, 4, 5]