如何通过引用它在熊猫中的日期来提取

Question

我有 df 喜欢跟随一个customer  movement      dateA         buy      2019/5/4 A        inquiry   2020/7/1A        cancel    2020/8/1B         buy      2019/6/1B        cancel    2020/8/1我想追查每个客户movement之前cancel首先，分组依据customerA         buy      2019/5/4 A        inquiry   2020/7/1A        cancel    2020/8/1然后我想得到cancel dateA        cancel    2020/8/1然后，我想previous movement在取消之前得到in 1 year。customer movement dateA        inquiry  2020/7/1A        cancel   2020/8/1在那之后，我想在每个customers所以我想要的结果如下customer movement dateA        inquiry  2020/7/1A        cancel   2020/8/1B        cancel   2020/8/1有什么办法可以做到这一点？这太复杂了，我无法处理这样的程序..

Answer 1

首先将列转换为日期时间并创建仅过滤行的cancel系列DataFrame.set_index：

df['date'] = pd.to_datetime(df['date'])

s = df[df['movement'].eq('cancel')].set_index('customer')['date']

然后按Series.map年减去 1 年进行映射，并按indate过滤列的较少值：Series.ltboolean indexing

df = df[df['customer'].map(s.sub(pd.DateOffset(years=1))).lt(df['date'])]

print (df)

  customer movement       date

1        A  inquiry 2020-07-01

2        A   cancel 2020-08-01

4        B   cancel 2020-08-01