Python – 使用pandas搜索DataFrame中的特定值

我们可以搜索DataFrame中的特定值。使用iloc获取所需值并显示整行。首先，导入所需的库 –

import pandas as pd

创建一个有4个列的DataFrame –

dataFrame = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],"Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],"Units_Sold": [ 100, 120, 150, 110, 200, 250]
})

让我们搜索注册价格为500的汽车 –

for i in range(len(dataFrame.Car)):
   if 5000 == dataFrame.Reg_Price[i]:
      indx = i

现在，显示找到的值 –

dataFrame.iloc[indx]

示例

以下是代码 –

import pandas as pd

# 创建DataFrame
dataFrame = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Tesla', 'Mustang', 'Mercedes', 'Jaguar'],"Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],"Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],"Units_Sold": [ 100, 120, 150, 110, 200, 250]
})

print"DataFrame ...\n",dataFrame

# 搜索注册价格为500的汽车
for i in range(len(dataFrame.Car)):
   if 5000 == dataFrame.Reg_Price[i]:
      indx = i

# 显示找到的值
print"\n搜索了以下特定值的DataFrame...\n",dataFrame.iloc[indx]

输出

这将产生以下输出 –

DataFrame ...
        Car   Cubic_Capacity   Reg_Price   Units_Sold
0       BMW             2000        7000          100
1     Lexus             1800        1500          120
2     Tesla             1500        5000          150
3   Mustang             2500        8000          110
4  Mercedes             2200        9000          200
5    Jaguar             3000        6000          250

搜索了以下特定值的DataFrame...
Car             Tesla
Cubic_Capacity   1500
Reg_Price        5000
Units_Sold        150
Name: 2, dtype: object