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

Python Pandas - Tìm các hàng chung giữa hai Khung dữ liệu

Để tìm các hàng chung giữa hai DataFrame, hãy sử dụng phương thức merge (). Đầu tiên chúng ta hãy tạo DataFrame1 với hai cột -

dataFrame1 = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90] }
)

Tạo DataFrame2 với hai cột -

dataFrame2 = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 250, 150, 80, 130, 90]
   }
)

Để tìm các hàng chung -

dataFrame1.merge(dataFrame2, how = 'inner' ,indicator=False)

Ví dụ

Sau đây là mã -

import pandas as pd

# Create DataFrame1
dataFrame1 = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Tesla', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90] }
)

print"DataFrame1 ...\n",dataFrame1

# Create DataFrame2
dataFrame2 = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 250, 150, 80, 130, 90]
   }
)

print"\nDataFrame2 ...\n",dataFrame2

# check for equality
print"\nAre both the DataFrames equal? ",dataFrame1.equals(dataFrame2)

# finding common rows between two DataFrames
resData = dataFrame1.merge(dataFrame2, how = 'inner' ,indicator=False)
print"\nCommon rows between two DataFrames...\n",resData

Đầu ra

Điều này sẽ tạo ra kết quả sau -

DataFrame1 ...
       Car   Units
0      BMW    100
1    Lexus    150
2     Audi    110
3    Tesla     80
4  Bentley    110
5   Jaguar     90

DataFrame2 ...
       Car   Units
0      BMW    100
1    Lexus    250
2     Audi    150
3  Mustang     80
4  Bentley    130
5   Jaguar     90

Are both the DataFrames equal? False

Common rows between two DataFrames...
      Car   Units
0     BMW    100
1  Jaguar     90