使用 Pandas 選擇包含特定文字的行


要選擇包含特定文字的行,請使用 contains() 方法。假設以下是我們 CSV 檔案的路徑 −

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

首先,讓我們讀取 CSV 檔案並建立 Pandas 資料框 −

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

現在,讓我們選擇包含特定文字“BMW”的行 −

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

範例

程式碼如下 −

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)

輸出

這將產生以下輸出 −

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

更新日期: 30-9 月 -2021

723 瀏覽次數

職業生涯一開始

透過完成課程獲得認證

開始
廣告