Chúng ta có thể thiết lập điều kiện và lấy hàng DataFrame. Các điều kiện này có thể được thiết lập bằng toán tử logic và cả toán tử quan hệ.
Trước tiên, nhập thư viện pandas:
import pandas as pd
Tạo DataFrame và đọc file CSV:
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")
Lấy hàng DataFrame với giá đăng ký nhỏ hơn 1000. Sử dụng toán tử quan hệ:
dataFrame[dataFrame.Reg_Price < 1000]
Ví dụ hoàn chỉnh
Mã nguồn đầy đủ:
import pandas as pd
# đọc file csv
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")
print("DataFrame...\n", dataFrame)
# đếm hàng và cột trong DataFrame
print("\nSố hàng và cột trong DataFrame = ", dataFrame.shape)
# lấy hàng DataFrame với giá đăng ký nhỏ hơn 1000
resData = dataFrame[dataFrame.Reg_Price < 1000]
print("DataFrame...\n", resData)
Kết quả
Chương trình 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