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)

更新於: 27-Sep-2021

180 檢視

開啟您的 事業

透過完成課程獲得認證

開始
廣告