Python Pandas – 查询DataFrame的列

要查询Pandas DataFrame的列，请使用query()。我们进行查询以过滤记录。首先，让我们创建一个Pandas DataFrame

dataFrame = pd.DataFrame({"Product": ["智能电视", "U盘", "音箱", "耳机"],"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": ["智能电视", "U盘", "音箱", "耳机"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

print("DataFrame...\n",dataFrame)

# 使用query()查询列
print("\n查询列以过滤记录...\n")
print(dataFrame.query('Opening_Stock >=500 & Closing_Stock < 1000 & Product.str.startswith("P").values'))

输出

这将生成以下输出−

DataFrame...
     Closing_Stock   Opening_Stock   Product
0             200             300    智能电视
1             500             700    U盘
2            1000            1200    音箱
3             900            1500    耳机

查询列以过滤记录...

    Closing_Stock   Opening_Stock   Product
1       500                 700     U盘