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)
广告