使用python根据日期条件删除行

debugcn 发表于 Dev

冠心病

我有一个数据框，其中包括日期和ID列。以下是仅出于此问题目的的示例框架。但是实际数据包括更多的行和列。

from datetime import date, timedelta
import pandas as pd

date = datetime.datetime(2020, 1, 1)
delta_1 = 5
delta_2 = 15
delta_3 = 18

data = {
    'A': [date, date - timedelta(delta_1), date - timedelta(delta_2), date, date - timedelta(delta_3)], 
    'B': ['a', 'a', 'a', 'b', 'b']
}
df = pd.DataFrame(data)
print(df)

           A  B
0 2020-01-01  a
1 2019-12-27  a
2 2019-12-17  a
3 2020-01-01  b
4 2019-12-14  b

我要实现的是，对于每个唯一ID（B在示例中为列），从最新的行开始，然后根据日期条件删除行：如果从最旧的ID开始的10天内插入了具有现有ID的行具有该ID的最近行，只有最新行才有效。因此，在此示例中，以10天为限制，我最终得到以下结果：

           A  B
0 2020-01-01  a
2 2019-12-17  a
3 2020-01-01  b
4 2019-12-14  b

任何想法将不胜感激！

贝尼

这是一种diff与结合使用的方法，cumsum可得到日差总和，然后通过//

s=df.groupby('B').A.apply(lambda x : x.diff().dt.days.cumsum().fillna(0).abs()//10)
df=df.groupby([df.B,s]).head(1)
           A  B
0 2020-01-01  a
2 2019-12-17  a
3 2020-01-01  b
4 2019-12-14  b

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。