Computer >> Máy Tính >  >> Lập trình >> Python

Python Pandas - Cách chọn các hàng DataFrame trên cơ sở các điều kiện

Chúng tôi có thể đặt điều kiện và tìm nạp các hàng DataFrame. Các điều kiện này có thể được đặt bằng cách sử dụng các toán tử logic và thậm chí cả các toán tử quan hệ.

Đầu tiên, hãy nhập các thư viện gấu trúc được yêu cầu -

import pandas as pd

Hãy để chúng tôi tạo DataFrame và đọc tệp CSV của chúng tôi -

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

Tìm nạp các hàng khung dữ liệu có giá đăng ký nhỏ hơn 1000. Chúng tôi đang sử dụng toán tử quan hệ cho việc này -

dataFrame[dataFrame.Reg_Price < 1000]

Ví dụ

Sau đây là mã -

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)

Đầu ra

Điều này sẽ tạo ra kết quả sau -

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