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

Python - Lọc dữ liệu với phương thức Pandas .query()


Pandas là một thư viện Python rất phổ biến dùng cho làm sạch dữ liệu, phân tích dữ liệu, v.v. Trong bài viết này chúng ta sẽ xem cách sử dụng phương thức query để lấy dữ liệu cụ thể từ một tập dữ liệu cho trước. Chúng ta có thể có cả điều kiện đơn lẻ và nhiều điều kiện bên trong một query.

Đọc dữ liệu

Hãy đọc dữ liệu vào một pandas DataFrame sử dụng thư viện pandas. Chương trình dưới đây làm chính điều đó.

Ví dụ

import pandas as pd
# Đọc DataFrame từ file csv
data = pd.read_csv("D:\\heart.csv")
print(data)

Đầu ra

Chạy mã trên cho kết quả sau −

Python - Lọc dữ liệu với phương thức Pandas .query()

Query với điều kiện đơn

Tiếp theo chúng ta xem cách sử dụng query với điều kiện đơn. Như bạn thấy chỉ 119 hàng từ 303 hàng gốc được trả về.

Ví dụ

import pandas as pd
# DataFrame từ file csv
data = pd.read_csv("D:\\heart.csv")
data.query('chol < 230', inplace=True)
# Kết quả
print(data)

Đầu ra

Chạy mã trên cho kết quả sau −

Python - Lọc dữ liệu với phương thức Pandas .query()

Query với nhiều điều kiện

Theo cách tương tự, chúng ta có thể áp dụng nhiều điều kiện cho phương thức query. Điều này sẽ giới hạn tập kết quả thêm. Chỉ 79 hàng được trả về khi chúng ta cũng giới hạn age > 60.

Ví dụ

import pandas as pd
# DataFrame từ file csv
data = pd.read_csv("D:\\heart.csv")
data.query('chol < 230 and age > 60', inplace=True)
# Kết quả
print(data)

Đầu ra

Chạy mã trên cho kết quả sau −

Python - Lọc dữ liệu với phương thức Pandas .query()