熊猫：一个数据框与另一数据框的列的匹配索引

Question

我有两个名为df1和df2的数据帧：

df1

df2

SITE_NUMBER被设置为df1的索引。现在，如果df2的'S_STATION'的值与df1的索引值匹配，我想返回df1的'FACILITY'列。我为此编写了一个函数：

def return_facility():
   return df1[df1['SITE_NUMBER'] == pd.to_numeric(df2['S_STATION'])]['FACILITY']

但是，该函数不起作用，因为编译器无法识别df1 ['SITE_NUMBER']。有人可以指出我在哪里犯错吗？

Answer 1

您需要isin。在df1.index

上使用它

def return_facility():
   return df1.loc[df1.index.isin(pd.to_numeric(df2['S_STATION'])), 'FACILITY']