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

Chọn dữ liệu với tiêu chí phức tạp từ DataFrame Pandas

Chúng ta có thể sử dụng nhiều tiêu chí khác nhau để so sánh tất cả các giá trị cột của một DataFrame Pandas. Các phép so sánh như df[col] < 5, df[col] == 10, v.v. có thể được thực hiện. Ví dụ, nếu sử dụng tiêu chí df[col] > 2, nó sẽ kiểm tra tất cả giá trị từ cột col và so sánh xem chúng có lớn hơn 2 hay không. Đối với mọi giá trị cột, nó sẽ trả về True nếu điều kiện đúng, ngược lại trả về False. Hãy xem ví dụ sau để hiểu rõ hơn.

Các bước thực hiện

  • Tạo dữ liệu bảng hai chiều, có thể thay đổi kích thước, tiềm năng dị chất df.
  • In DataFrame đầu vào df.
  • Khởi tạo biến col với tên cột.
  • Thực hiện các phép so sánh.
  • In DataFrame kết quả.

Ví dụ

import pandas as pd

df = pd.DataFrame({
    "x": [5, 2, 7, 0],
    "y": [4, 7, 5, 1],
    "z": [9, 3, 5, 1]
})

print("Input DataFrame is:\n", df)

col = "x"
print("Phần tử > 5 trong cột", col, ":\n", df[col] > 5)
print("Phần tử == 5 trong cột", col, ":\n", df[col] == 5)

col = "y"
print("Phần tử < 5 trong cột", col, ":\n", df[col] < 5)
print("Phần tử != 5 trong cột", col, ":\n", df[col] != 5)

Kết quả đầu ra

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

Phần tử > 5 trong cột x:
0    False
1    False
2     True
3    False
Name: x, dtype: bool

Phần tử == 5 trong cột x:
0     True
1    False
2    False
3    False
Name: x, dtype: bool

Phần tử < 5 trong cột y:
0     True
1    False
2    False
3     True
Name: y, dtype: bool

Phần tử != 5 trong cột y:
0     True
1     True
2    False
3     True
Name: y, dtype: bool