使用 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
廣告