python Pandas : How to calculate elapsed time between spans?

Question

我有一个包含时间戳事件的 pandas DataFrame。每个事件都有开始时间和结束时间:

start  end other_vars
  100  120  ...
  150  151  ...
  160  170  ...
  200  210  ...

是否有一种干净的方法来计算 pandas 中事件之间的时间(例如，上一个事件的 结束 与此事件的 开始 之间的跨度)？

start  end between other_vars
  100  120      NA   ...
  150  151      30   ...
  160  170       9   ...
  200  210      30   ...

Answer 1

我认为实现这一点的最简单方法是从另一列中减去一个移位的列。移位函数正是这样做的，它默认将数组移动一个索引。

In [3]: df
Out[3]:
   start  end
0    100  120
1    150  151
2    160  170
3    200  210

In [4]: df.start - df.end.shift()
Out[4]:
0   NaN
1    30
2     9
3    30

In [5]: df['elapsed'] = df.start - df.end.shift()

In [6]: df
Out[6]:
   start  end  elapsed
0    100  120      NaN
1    150  151       30
2    160  170        9
3    200  210       30