How to merge same columns from two different DataFrames

Question

I have two Dataframes A and B. Both have same 4 columns. I want to merge the two data frames such that if first three column values match, then merge the id values(which is a jasonb array)

Sample data:

df_A

name     age    zip      id
abc      25     11111    ["2722", "2855", "3583"]

df_B

name     age    zip      id
abc      25     11111    ["123", "234"]

I want the final output to look like

Final output:

name     age    zip      id
----------------------------------------------------------------
abc      25     11111    ["2722", "2855", "3583", "123", "234"]

BENY · Accepted Answer · 2019-06-19 03:15:43Z

1

One quick solution will be

l=['name','age','zip']
df=(df1.set_index(l)+df2.set_index(l)).reset_index()

answered Jun 19, 2019 at 3:15

BENY

324k22 gold badges176 silver badges250 bronze badges

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

Comments

cs95 · Accepted Answer · 2019-06-19 03:17:19Z

1

Another option is to merge, then use a list comprehension to handle the "id" columns.

output = df_A.merge(df_B, on=['name', 'age', 'zip'])
output['id'] = [[*x, *y] for x, y in zip(output.pop('id_x'), output.pop('id_y'))] 

output
  name  age    zip                            id
0  abc   25  11111  [2722, 2855, 3583, 123, 234]

answered Jun 19, 2019 at 3:17

cs95

406k106 gold badges744 silver badges797 bronze badges

Collectives™ on Stack Overflow

How to merge same columns from two different DataFrames

Sample data:

df_A

df_B

Final output:

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

Sample data:

df_A

df_B

Final output:

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related