[python] 파이썬 pandas에서 데이터를 필터링하는 방법은 무엇인가요?

20 Dec 2023

python

import pandas as pd

# 샘플 데이터프레임 생성
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Los Angeles', 'Chicago', 'Houston']}
df = pd.DataFrame(data)

# Age 열의 값이 30 이상인 행을 필터링
filtered_df = df[df['Age'] >= 30]
print(filtered_df)

위 코드에서 df['Age'] >= 30는 Age 열의 값이 30 이상인지 여부에 따라 True/False 값을 갖는 시리즈를 생성합니다. 이를 다시 데이터프레임에 전달하여 해당 조건에 맞는 행만 필터링하여 출력할 수 있습니다.

판다스에서 데이터를 필터링하는 다양한 방법이 있으며, 쿼리 메서드나 query() 함수를 사용하여 SQL 스타일의 데이터베이스 쿼리를 수행하는 방법도 있습니다. 자세한 내용은 판다스 공식 문서를 참고하시기 바랍니다. 판다스 공식 문서