Получите разницу двух столбцов со смещением / вращением / смещением 1

Question

Глупый вопрос: у меня есть два столбца A и B, и я хотел бы создать new_col, что на самом деле является разницей между текущим B и предыдущим A. Предыдущий, например, означает строку перед текущей строкой. Как этого достичь (возможно, даже с переменным смещением)?

Цель:

df
| A | B  | new_col  |
|---|----|----------|
| 1 | 2  | nan (or2)|
| 3 | 4  | 3        |
| 5 | 10 | 7        |

Псевдокод:

new_col[0] = B[0] - 0
new_col[1] = B[1] - A[0]
new_col[2] = B[2] - A[1]

Answer 1

Использование Series.shift:

df['new_col'] = df['B'] - df['A'].shift()

   A   B  new_col
0  1   2      NaN
1  3   4      3.0
2  5  10      7.0