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”的车列−
for k in dataFrame.itertuples():
if k[1].find('Lamborghini') != -1:
print(k)
示例
以下是代码
import pandas as pd
# 读取CSV文件
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")
print("DataFrame...\n",dataFrame)
# 迭代并获取包含特定文本的行
# 我们正在查找带有文本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)
极客教程