PANDAS - разница между значениями в разных строках и разных столбцах

Question

Я хотел бы рассчитать разницу между прибытием и следующим отправлением ...
С:

Dep Arr
1 2
4 6
8 9

До

Dep Arr BreakAfter
1 2 2
4 6 2
8 9 Nan

... в более быстрый способ, чем цикл for.
Есть ли что-то вроде .diff (), которое может работать с разными столбцами?

Answer 1

Вы можете использовать shift.

Это будет нормально работать:

In [274]: df                                                                                                                                                                                                
Out[274]: 
   Dep  Arr
0    1    2
1    4    6
2    8    9

In [280]: df['Dep'].shift(-1) - df['Arr']                                                                                                                                                                         
Out[280]: 
0    2.0
1    2.0
2    NaN
dtype: float64