Python Pandas - 遍歷並獲取包含所需文字的行
要遍歷並獲取包含所需文字的行,請使用 itertuples() 方法和 find() 方法。itertuples() 會遍歷 DataFrame 行。
首先,讓我們匯入所需庫並用別名表示 −
import pandas as pd
我們的 CSV 位於桌面,如下所示 −
C:\Users\amit_\Desktop\CarRecords.csv
我們來讀取 CSV 檔案並建立 Pandas DataFrame −
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")
遍歷並獲取包含特定文字的行。我們用“Lamborghini”文字獲取 Car 列 −
for k in dataFrame.itertuples(): if k[1].find('Lamborghini') != -1: print(k)
示例
以下為程式碼
import pandas as pd # reading csv file dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv") print("DataFrame...\n",dataFrame) # iterate and fetch the rows containing a specific text # we are finding for Car column with text Lamborghini for k in dataFrame.itertuples(): if k[1].find('Lamborghini') != -1: print(k)
輸出
將產生以下輸出 −
Pandas(Index=5, Car='Lamborghini', Place='Chandigarh', UnitsSold=80) Pandas(Index=8, Car='Lamborghini', Place='Delhi', UnitsSold=100)
廣告