Python Pandas在2个数据框的行索引和列索引上合并

时间:2020-07-20 02:40:44

标签: python pandas dataframe merge

我正在尝试执行以下操作:我计算一个新数据框,即dataframe1除以dataframe2,其中columnname匹配和date索引基于最近的日期(不完全匹配)进行匹配

idx1 = pd.DatetimeIndex(['2017-01-01','2018-01-01','2019-01-01'])
idx2 = pd.DatetimeIndex(['2017-02-01','2018-03-01','2019-04-01'])
df1 = pd.DataFrame(index = idx1,data = {'XYZ': [10, 20, 30],'ABC': [15, 25, 30]})
df2 = pd.DataFrame(index = idx2,data = {'XYZ': [1, 2, 3],'ABC': [3, 5, 6]})

#looking for some code
#df3 = df1/df2 on matching column and closest matching row

这应该产生一个看起来像这样的数据框

           XYZ  ABC
2017-01-01  10  5
2018-01-01  10  5
2019-01-01  10  5

1 个答案:

答案 0 :(得分:0)

您可以使用asof合并对“关闭”行进行匹配。然后,我们将对列轴进行分组并进行划分。

df3 = pd.merge_asof(df1, df2, left_index=True, right_index=True,
                    direction='nearest')
#            XYZ_x  ABC_x  XYZ_y  ABC_y
#2017-01-01     10     15      1      3
#2018-01-01     20     25      2      5
#2019-01-01     30     30      3      6

df3 = (df3.groupby(df3.columns.str.split('_').str[0], axis=1)
           .apply(lambda x: x.iloc[:, 0]/x.iloc[:, 1]))
#            ABC   XYZ
#2017-01-01  5.0  10.0
#2018-01-01  5.0  10.0
#2019-01-01  5.0  10.0