Để lọc hàng và lấy giá trị cột cụ thể, sử dụng phương thức contains() của Pandas. Trước tiên, nhập thư viện cần thiết với bí danh:
import pandas as pd
Đọc file CSV bằng read_csv(). File CSV trên Desktop:
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv")
Bây giờ, lọc hàng với văn bản cụ thể:
dataFrame = dataFrame[dataFrame['Car'].str.contains('Lamborghini')]
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\\CarRecords.csv")
print("DataFrame...\n", dataFrame)
# chọn hàng chứa văn bản "Lamborghini"
dataFrame = dataFrame[dataFrame['Car'].str.contains('Lamborghini')]
print("\nLấy hàng với văn bản Lamborghini ...\n", dataFrame)
Kết quả
Chương trình sẽ tạo ra kết quả sau:
DataFrame... Car Place UnitsSold 0 Audi Bangalore 80 1 Porsche Mumbai 110 2 RollsRoyce Pune 100 3 BMW Delhi 95 4 Mercedes Hyderabad 80 5 Lamborghini Chandigarh 80 6 Audi Mumbai 100 7 Mercedes Pune 120 8 Lamborghini Delhi 100 Fetching rows with text Lamborghini ... Car Place UnitsSold 5 Lamborghini Chandigarh 80 8 Lamborghini Delhi 100