Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

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