Pandas датафрейм, как добавить столбец расстояния от предыдущего ряда?

Question

У меня есть данные о местах:

df = X    Y 
     1    1
     2    1
     2    1 
     2    2
     3    3
     5    5    
    5.5   5.5  

Я хочу добавить столбцы с расстоянием до предыдущей точки: Так будет:

df = X    Y    Distance
     1    1     0
     2    1     1
     2    1     0
     2    2     1
     3    3     2
     5    5     2
    5.5   5.5   1

Что такое лучший способ сделать это?

Answer 1

Вы можете использовать метод pd.Series.diff .

Например, чтобы вычислить эйлерово расстояние, используя также np.sqrt , вы должны сделать так:

import numpy as np
df["Distance"] = np.sqrt(df.X.diff()**2 + df.Y.diff()**2)