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

Làm cách nào để tìm các phần tử chung trong Pandas DataFrame?

Để tìm các phần tử phổ biến trong Pandas DataFrame, chúng ta có thể sử dụng phương thức merge () với danh sách các cột

Các bước

  • Tạo dữ liệu dạng bảng hai chiều, có thể thay đổi kích thước, có khả năng không đồng nhất, df1 .

  • In DataFrame đầu vào, df1 .

  • Tạo dữ liệu dạng bảng hai chiều khác, df2 .

  • In DataFrame đầu vào, df2 .

  • Tìm các phần tử phổ biến bằng cách sử dụng merge () phương pháp.

  • 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("Input DataFrame 1 is:\n", df1)
print("Input DataFrame 2 is:\n", df2)
common = df1.merge(df2, on=['x', 'y', 'z'])
print("Common of DataFrame 1 and 2 is: \n", common)

Đầu ra

Input DataFrame 1 is:
   x  y  z
0  5  4  9
1  2  7  3
2  7  5  5
3  0  1  1

Input DataFrame 2 is:
   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
Common of DataFrame 1 and 2 is:
   x  y z
0  5  4 9
1  2 7 3
2  7 5 5
3  0 1 1