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

Python Pandas - Chúng ta có thể sử dụng &Operator để tìm các cột chung giữa hai DataFrame không?

Có, chúng ta có thể sử dụng toán tử &để tìm các cột chung giữa hai DataFrame. Đầu tiên, chúng ta hãy tạo hai DataFrames -

# creating dataframe1
dataFrame1 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],
})

print("Dataframe1...\n",dataFrame1)

# creating dataframe2
dataFrame2 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Units_Sold": [ 100, 110, 150, 80, 200, 90]
})

Nhận các cột chung bằng cách sử dụng toán tử &-

res = dataFrame1.columns & dataFrame2.columns

Ví dụ

Sau đây là mã -

import pandas as pd

# creating dataframe1
dataFrame1 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],
})

print"Dataframe1...\n",dataFrame1

# creating dataframe2
dataFrame2 = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Units_Sold": [ 100, 110, 150, 80, 200, 90]
})

print"Dataframe2...\n",dataFrame2

# getting common columns using the & operator
res = dataFrame1.columns & dataFrame2.columns

print"\nCommon columns...\n",res

Đầu ra

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

Dataframe1...
        Car   Cubic_Capacity
0       BMW             2000
1     Lexus             1800
2     Tesla             1500
3   Mustang             2500
4  Mercedes             2200
5    Jaguar             3000
Dataframe2...
        Car   Units_Sold
0       BMW          100
1     Lexus          110
2     Tesla          150
3   Mustang           80
4  Mercedes          200
5    Jaguar           90

Common columns...
Index([u'Car'], dtype='object')