Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

Python Pandas – วิธีเลือกแถว DataFrame ตามเงื่อนไข


เราสามารถกำหนดเงื่อนไขและดึงข้อมูลแถว DataFrame เงื่อนไขเหล่านี้สามารถตั้งค่าได้โดยใช้ตัวดำเนินการเชิงตรรกะและแม้แต่ตัวดำเนินการเชิงสัมพันธ์

ขั้นแรก นำเข้าไลบรารีแพนด้าที่จำเป็น -

import pandas as pd

ให้เราสร้าง DataFrame และอ่านไฟล์ CSV ของเรา -

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")

กำลังดึงแถว dataframe ที่มีราคาลงทะเบียนน้อยกว่า 1,000 เรากำลังใช้ตัวดำเนินการเชิงสัมพันธ์สำหรับสิ่งนี้ -

dataFrame[dataFrame.Reg_Price < 1000]

ตัวอย่าง

ต่อไปนี้เป็นรหัส -

import pandas as pd

# reading csv file
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")
print("DataFrame...\n",dataFrame)

# count the rows and columns in a DataFrame
print("\nNumber of rows and column in our DataFrame = ",dataFrame.shape)

# fetching dataframe rows with registration price less than 1000
resData = dataFrame[dataFrame.Reg_Price < 1000]

print("DataFrame...\n",resData)

ผลลัพธ์

สิ่งนี้จะสร้างผลลัพธ์ต่อไปนี้ -

DataFrame...
           Car   Date_of_Purchase   Reg_Price
0          BMW         10/10/2020        1000
1        Lexus         10/12/2020         750
2         Audi         10/17/2020         750
3       Jaguar         10/16/2020        1500
4      Mustang         10/19/2020        1100
5  Lamborghini         10/22/2020        1000

Number of rows and column in our DataFrame = (6, 3)
DataFrame...
     Car   Date_of_Purchase   Reg_Price
1  Lexus         10/12/2020         750
2   Audi         10/17/2020         750