Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Lọc hàng - Python Pandas

Để 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