Chọn các hàng có chứa văn bản cụ thể bằng cách sử dụng Pandas

Để chọn các hàng có chứa văn bản cụ thể, hãy sử dụng phương thức chứa (). Giả sử sau đây là đường dẫn tệp CSV của chúng tôi -

C:\\Users\\amit_\\Desktop\\SalesRecords.csv

Đầu tiên, hãy để chúng tôi đọc tệp CSV và tạo Pandas DataFrame -

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

Bây giờ, chúng ta hãy chọn các hàng có chứa văn bản cụ thể “BMW” -

dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]

Ví dụ

Sau đây là mã -

import pandas as pd

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

# select rows containing text "BMW"
dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]
print("\nFetching rows with text BMW ...\n",dataFrame)

Đầu ra

Điều này 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 BMW ...
   Car   Place UnitsSold
3  BMW   Delhi 95