Python Pandas - Truy vấn các cột của DataFrame

Để truy vấn các cột của Pandas DataFrame, hãy sử dụng truy vấn (). Chúng tôi đang truy vấn để lọc các bản ghi. Đầu tiên, chúng ta hãy tạo một Pandas DataFrame

dataFrame = pd.DataFrame({"Product": ["SmartTV", "PenDrive", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

Sử dụng query () để truy vấn các cột có điều kiện -

print(dataFrame.query('Opening_Stock >=500 & Closing_Stock < 1000 & Product.str.startswith("P").values'))

Ví dụ

Sau đây là mã hoàn chỉnh -

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["SmartTV", "PenDrive", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

print"DataFrame...\n",dataFrame

# using query() to query columns
print"\nQuerying columns to filter records...\n"
print(dataFrame.query('Opening_Stock >=500 & Closing_Stock < 1000 & Product.str.startswith("P").values'))

Đầu ra

Điều này sẽ tạo ra kết quả sau -

DataFrame...
     Closing_Stock   Opening_Stock   Product
0             200             300    SmartTV
1             500             700    PenDrive
2            1000            1200    Speaker
3             900            1500    Earphone

Querying columns to filter records...

    Closing_Stock   Opening_Stock   Product
1       500                 700     PenDrive