Запросы dataframe на основе численного сходства между рядами

Question

у меня есть dataframe вроде этого:

 Allotment  Date   NDVI_Kurtosis NDVI_Skewness 
1 D    19840621  1.02    3.06 
2 D    19850619  1.76    2.56 
3 A    19840621  3.66    3.50 
4 A    19850619  1.56    3.20 

и я хочу, чтобы вернуться каждый Allotment и связанный с Date если ОБЕ NDVI_Kurtosis и NDVI_Skewness находятся в пределах 1,00 друг от друга между различными строк. Таким образом, в этом случае, я хотел бы это вернулся:

Allotment   Date   NDVI_Kurtosis NDVI_Skewness 
    D    19840621  1.02    3.06 
    D    19850619  1.76    2.56 
    A    19850619  1.56    3.20 

Я играл вокруг с помощью iloc для этого, но не увенчались успехом до сих пор.

Answer 1

Вы можете использовать функцию сдвига для создания новых столбцов и после того, как вы сможете сравнить их с начальными столбцами.

import pandas as pd 
df=pd.read_clipboard() 
df['NDVI_Kurtosis_lag']=df['NDVI_Kurtosis'].shift(1).fillna(df['NDVI_Kurtosis']) 
df['NDVI_Skewness_lag']=df['NDVI_Skewness'].shift(1).fillna(df['NDVI_Skewness']) 
df 
df2=df[(df['NDVI_Kurtosis']-df['NDVI_Kurtosis_lag']<1) & (df['NDVI_Skewness']-df['NDVI_Skewness_lag']<1)] 
df2 

Cordialy, Laurent.