使用Pandas选择包含特定文本的行

要选择包含特定文本的行，请使用contains()方法。假设以下是我们的CSV文件路径：

C:\Users\amit_\Desktop\SalesRecords.csv

首先，让我们读取CSV文件并创建Pandas DataFrame –

dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")

现在，让我们选择包含特定文本“BMW”的行 –

dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]

示例

以下是代码 –

import pandas as pd

# 读取csv文件
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")
print("DataFrame...\n",dataFrame)

# 选择包含文本“BMW”的行
dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]
print("\n获取带有BMW文本的行...\n",dataFrame)

输出

这将产生以下输出 –

DataFrame ...
 Car       Place   UnitsSold
0         Audi   Bangalore          80
1      Porsche      Mumbai         110
2   RollsRoyce        Pune         100
3          BMW       Delhi          95
4     Mercedes   Hyderabad          80
5  Lamborghini  Chandigarh          80
6         Audi      Mumbai         100
7     Mercedes        Pune         120
8  Lamborghini       Delhi         100

获取带有BMW文本的行 ...
 Car   Place UnitsSold
3  BMW   Delhi 95