合并特定列和特定数据框的功能

Question

我正在尝试创建一个函数来合并特定列上的两个数据框，如下所示：

list_to_be_merged_on = ['code', 'name', 'section']

def merge_on_specific_columns(table1, table2, list_to_be_merged_on):
     merged_table = pd.merge(table1, table2, on=[list_to_be_merged_on])
     return merged_table


 merged_df = 
 merge_on_specific_columns(file,car,list_to_be_merged_on)

我收到以下错误：

KeyError: ('code', 'name', 'section')

对可能出现的问题有什么建议吗？

Answer 1

您不需要将 list_to_be_meged_on 放在列表中，因为它已经是一个列表。
尝试“on=list_to_be_merged_on”而不是“on=[list_to_be_merged_on]”