选择 MultiIndex 在另一个 DataFrame 中的行

Question

我有一个带有 MultiIndex 和许多附加列的 DataFrame (DF1)。在另一个 DataFrame (DF2) 中，我有 2 列包含来自 MultiIndex 的一组值。我想从 DF1 中选择 MultiIndex 与 DF2 中的值匹配的行。

df1 = pd.DataFrame({'month': [1, 3, 4, 7, 10],
                    'year': [2012, 2012, 2014, 2013, 2014],
                    'sale':[55, 17, 40, 84, 31]})

df1 = df1.set_index(['year','month'])

            sale
year  month
2012  1     55
2012  3     17
2014  4     40
2013  7     84
2014  10    31

df2 = pd.DataFrame({'year': [2012,2014],
                    'month': [1, 10]})

   year  month  
0  2012   1
1  2014   10

我想创建一个新的 DataFrame：

                 sale
year    month
2012    1        55
2014    10       31

我尝试了许多使用 .isin、.loc、切片的变体，但总是遇到错误。

Answer 1

您可以以同样的方式在df2 上使用set_index 并传递索引：

In[110]:
df1.loc[df2.set_index(['year','month']).index]

Out[110]: 
            sale
year month      
2012 1        55
2014 10       31

更易读的版本：

In[111]:
idx = df2.set_index(['year','month']).index
df1.loc[idx]

Out[111]: 
            sale
year month      
2012 1        55
2014 10       31