Python Pandas – Có thể dùng toán tử & để tìm cột chung giữa hai DataFrame?

Có, ta có thể dùng toán tử & để tìm cột chung giữa hai DataFrame.

Trước hết, tạo hai DataFrame:

# Tạo dataframe1
dataFrame1 = pd.DataFrame({
    "Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],
    "Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],
})
print("Dataframe1...\n", dataFrame1)

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

Lấy cột chung dùng toán tử &:

res = dataFrame1.columns & dataFrame2.columns

Ví dụ hoàn chỉnh

import pandas as pd

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

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

res = dataFrame1.columns & dataFrame2.columns
print("\nCommon columns...\n", res)

Đầu ra

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(['Car'], dtype='object')