Function to merge on specific columns and specific dataframes

Question

I am trying to create a function to merge two dataframes on specific columns as follows:

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)

I get the following error:

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

any suggestions to what might go wrong?

Wightboy · Accepted Answer · 2021-10-12 13:48:05Z

1

You don't need to put list_to_be_meged_on in a list as it is already a list.
Try "on=list_to_be_merged_on" instead of "on=[list_to_be_merged_on]"

answered Oct 12, 2021 at 13:48

Wightboy

2611 silver badge12 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Function to merge on specific columns and specific dataframes

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related