Computer >> Máy Tính >  >> Lập trình >> Python

Viết chương trình bằng Python để hợp nhất hai khung dữ liệu dựa trên dữ liệu khớp trong một cột

Giả sử, bạn có hai khung dữ liệu,

first dataframe is
 id country
0 1 India
1 2 UK
2 3 US
3 4 China

second dataframe is

 id City
0 1 Chennai
1 11 Cambridge
2 22 Chicago
3 4 Chengdu

Và kết quả để hợp nhất dựa trên cùng một cột là,

Merging data based on same column - id
 id    country    City
0 1    India    Chennai
1 4    China    Chengdu

Giải pháp

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước được đưa ra bên dưới -

  • Xác định hai khung dữ liệu

  • Hợp nhất hai khung dữ liệu dựa trên cùng một id cột được xác định bên dưới,

pd.merge(first_df,second_df,how='inner',on='id')

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau đây để hiểu rõ hơn -

import pandas as pd
import numpy as np
first_df = pd.DataFrame({'id':[1,2,3,4], 'country':['India','UK','US','China']
                        })
print("first dataframe is\n", first_df)
second_df = pd.DataFrame({'id':[1,11,22,4], 'City' : ['Chennai','Cambridge','Chicago','Chengdu']
                        })
print("second dataframe is\n", second_df)
print("Merging data based on same column - id")
print(pd.merge(first_df,second_df,how='inner',on='id'))

Đầu ra

first dataframe is
 id country
0 1 India
1 2 UK
2 3 US
3 4 China

second dataframe is
  id  City
0 1 Chennai
1 11 Cambridge
2 22 Chicago
3 4 Chengdu

Merging data based on same column - id
 id country City
0 1 India Chennai
1 4 China Chengdu