Python Pandas - 查詢 DataFrame 的列


要查詢 Pandas DataFrame 的列,請使用 query()。我們查詢是為了過濾記錄。首先,讓我們建立一個 Pandas DataFrame

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

使用 query() 根據條件查詢列 −

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

範例

以下為完整程式碼 −

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'))

輸出

這將產生以下輸出 −

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

更新於: 2021-9-14

334 次瀏覽

開啟你的職業生涯 旅程

完成課程,獲得認證

開始行動
廣告
© . All rights reserved.