Python - Tìm kiếm DataFrame cho giá trị cụ thể với pandas

Ta có thể tìm kiếm DataFrame cho giá trị cụ thể. Dùng iloc lấy giá trị yêu cầu và hiển thị toàn bộ hàng.

Nhập thư viện:

import pandas as pd

Tạo DataFrame 4 cột:

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]
})

Tìm Car với Registration Price 5000:

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

Hiển thị giá trị tìm thấy:

dataFrame.iloc[indx]

Ví dụ

import pandas as pd

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)

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

print("\nSearched DataFrame for the following specific value...\n", dataFrame.iloc[indx])

Đầu ra

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

Searched DataFrame for the following specific value...
Car                 Tesla
Cubic_Capacity      1500
Reg_Price           5000
Units_Sold           150
Name: 2, dtype: object

Lưu ý: Cách hiệu quả hơn là dùng boolean indexing: dataFrame[dataFrame['Reg_Price'] == 5000].