python - Pandas 等效于 SQL 非 equi JOIN

Question

所以我有 2 个数据框，我想合并在一起。

我正在合并 3 列，2 列很容易连接。

joined_df = pd.merge(df1, df2, how='left', on=['name', 'city'])

我希望它使用第三列，但这将是一个比较，如下所示:

joined_df = pd.merge(df1, df2, how='left',
on=['name', 'city', 'df1.year' >= 'df2.year_min'])

不确定这里的正确语法是什么。

如果是 SQL，对我来说会很容易。

SELECT * FROM df1
JOIN df2 on (df1.name = df2.name and df1.year = df2.year and df1.year > df2.year_min)

有什么帮助吗？

Answer 1

Pandas merge 只支持等值连接。您需要添加第二个步骤来过滤结果，如下所示:

joined_df = df1.merge(df2, how='left', on=['name', 'city'])
joined_df = joined_df[joined_df.year > joined_df.year_min]