Python Pandas - เลือก subset ของแถวและคอลัมน์ร่วมกัน

บทนำ

ใช้ .loc[] เลือกแถวและคอลัมน์พร้อมกัน ตั้งเงื่อนไขแถวและระบุคอลัมน์ที่ต้องการ

ขั้นตอน

1. อ่าน SalesData.csv
2. เลือกแถวที่ Units > 100 และแสดงเฉพาะคอลัมน์ Car:

   dataFrame.loc[dataFrame["Units"] > 100, "Car"]

โค้ดเต็ม

import pandas as pd

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n", dataFrame)

# แสดงรถที่ Units > 100
print("\nSelect cars with Units more than 100: \n", dataFrame[dataFrame["Units"] > 100])

# แสดง 2 คอลัมน์
res = dataFrame[['Reg_Price', 'Units']]
print("\nDisplaying only two columns : \n", res)

# Subset แถวและคอลัมน์ร่วมกัน
res2 = dataFrame.loc[dataFrame["Units"] > 100, "Car"]
print("\nSubset...\n", res2)

ผลลัพธ์

ได้ Series: Audi และ Mustang (Units > 100)