pandas Dataframe の複雑な行抽出

Question

以下のような条件があった時に、

1. DataFrameAから、DataFrameBのend_dateとDateが一致する行を特定
2. その行からnum行前の行を特定してそれらの行を新たなDataFrameとして抽出

としたい場合、どのような書き方があるでしょうか。
for文必須でしょうか。。？

num = 2

DataFrame A

Date	col1	col2
2019-01-05	aaa	bbb
2019-01-06	aaa	bbb
2019-01-07	aaa	bbb
2019-01-08	aaa	bbb
2019-01-09	aaa	bbb
2019-01-10	aaa	bbb
2019-01-11	aaa	bbb

DataFrame B

end_Date	col1	col2
2019-01-08	aaa	bbb
2019-01-11	aaa	bbb

算出したいDataframe

Date	col1	col2
2019-01-06	aaa	bbb
2019-01-09	aaa	bbb

Answer 1

ret = df_A[df_A['Date'].isin(df_B['end_Date']).shift(-num).fillna(False)]

でどうでしょうか

簡単に説明

1. Series.isin() を使って df_A['Date']の値がdf_B['end_Date'] の値を含まれているかを Boolean Listで取得 : df_A['Date'].isin(df_B['end_Date'])
2. 上記のBoolean Listをnum分だけ上にshift() する : df_A['Date'].isin(df_B['end_Date']).shift(-num)
3. shift() の結果シフト分だけ NaNが入るので fillna(False)する : df_A['Date'].isin(df_B['end_Date']).shift(-num).fillna(False)
4. 3のBooleanListを使ってdf_Aからフィルタリングする

以上

Comments

[tob]

まさしく求めていた回答でした。ありがとうございました！