Python - การกรองข้อมูลด้วยเมธอด .query() ของ Pandas

Pandas เป็นไลบรารี Python ที่ใช้กันอย่างกว้างขวางสำหรับการทำความสะอาดข้อมูล การวิเคราะห์ข้อมูล เป็นต้น ในบทความนี้เราจะดูวิธีใช้เมธอด query เพื่อดึงข้อมูลเฉพาะจากชุดข้อมูลที่กำหนด เราสามารถมีเงื่อนไขเดียวหรือหลายเงื่อนไขภายใน query ได้

การอ่านข้อมูล

มาอ่านข้อมูลเข้าไปใน pandas DataFrame โดยใช้ไลบรารี pandas โปรแกรมด้านล่างทำแค่นี้

ตัวอย่าง

import pandas as pd
# อ่าน DataFrame จากไฟล์ csv
data = pd.read_csv("D:\\heart.csv")
print(data)

Output

การรันโค้ดข้างต้นให้ผลลัพธ์ดังนี้ -

Query ด้วยเงื่อนไขเดียว

ต่อไปเราจะดูวิธีใช้เมธอด query ด้วยเงื่อนไขเดียว ตามที่เห็น มีเพียง 119 แถวจาก 303 แถวเดิมที่ถูกคืนค่าเป็นผลลัพธ์

ตัวอย่าง

import pandas as pd
# DataFrame จากไฟล์ csv
data = pd.read_csv("D:\\heart.csv")
data.query('chol < 230', inplace=True)
# ผลลัพธ์
print(data)

Output

การรันโค้ดข้างต้นให้ผลลัพธ์ดังนี้ -

Query ด้วยหลายเงื่อนไข

ด้วยวิธีการคล้ายกัน เราสามารถใช้หลายเงื่อนไขกับเมธอด query ได้ ซึ่งจะจำกัดชุดข้อมูลผลลัพธ์เพิ่มเติม ตอนนี้มีเพียง 79 แถวที่คืนค่าเมื่อเรากำหนดอายุมากกว่า 60 ด้วย

ตัวอย่าง

import pandas as pd
# DataFrame จากไฟล์ csv
data = pd.read_csv("D:\\heart.csv")
data.query('chol < 230 and age > 60', inplace=True)
# ผลลัพธ์
print(data)

Output

การรันโค้ดข้างต้นให้ผลลัพธ์ดังนี้ -