Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách tìm phần tử chung trong Pandas DataFrame


Để tìm các phần tử chung (hàng trùng khớp) giữa hai Pandas DataFrame, ta sử dụng phương thức merge() với danh sách cột.

Các bước

  • Tạo DataFrame đầu tiên df1.

  • In df1.

  • Tạo DataFrame thứ hai df2.

  • In df2.

  • Tìm phần tử chung bằng merge() trên các cột chung.

  • In DataFrame chung.

Ví dụ

import pandas as pd
df1 = pd.DataFrame({
    "x": [5, 2, 7, 0],
    "y": [4, 7, 5, 1],
    "z": [9, 3, 5, 1]
})
df2 = pd.DataFrame({
    "x": [5, 2, 7, 0, 11, 12],
    "y": [4, 7, 5, 1, 19, 20],
    "z": [9, 3, 5, 1, 29, 30]
})
print("DataFrame 1:\n", df1)
print("DataFrame 2:\n", df2)
common = df1.merge(df2, on=['x', 'y', 'z'])
print("Phần tử chung của DataFrame 1 và 2:\n", common)

Kết quả

DataFrame 1:
   x  y  z
0  5  4  9
1  2  7  3
2  7  5  5
3  0  1  1
DataFrame 2:
    x   y   z
0   5   4   9
1   2   7   3
2   7   5   5
3   0   1   1
4  11  19  29
5  12  20  30
Phần tử chung của DataFrame 1 và 2:
   x  y  z
0  5  4  9
1  2  7  3
2  7  5  5
3  0  1  1