使用 Python Pandas 过滤行

要过滤行并获取特定列的值，请使用 Pandas 的 contains() 方法。首先，让我们用别名导入所需的库−

import pandas as pd

使用 read_csv() 读取 CSV 文件。我们的 CSV 文件在桌面上−

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

现在，让我们过滤带有特定文本的行−

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

例子

以下是代码

import pandas as pd

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

# 选择包含文本“Lamborghini”的行
dataFrame = dataFrame[dataFrame['Car'].str.contains('Lamborghini')]
print("\n获取包含文本 Lamborghini 的行...\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

获取包含文本 Lamborghini 的行...
           Car       Place   UnitsSold
5  Lamborghini  Chandigarh          80
8  Lamborghini       Delhi         100