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 查看

开启您的 事业

通过完成课程获得认证

开始
广告