> 백엔드 개발 > 파이썬 튜토리얼 > 값 목록을 기반으로 Pandas DataFrame 행을 필터링하는 방법은 무엇입니까?

값 목록을 기반으로 Pandas DataFrame 행을 필터링하는 방법은 무엇입니까?

Linda Hamilton
풀어 주다: 2024-12-08 20:24:12
원래의
421명이 탐색했습니다.

How to Filter Pandas DataFrame Rows Based on a List of Values?

세트 목록의 값으로 DataFrame 행 필터링

Python pandas에서는 특정 열의 값이 있는지 여부에 따라 행을 필터링할 수 있습니다. 특정 값 집합에 존재합니다. 이를 달성하려면 다음과 같이 isin 메소드를 활용할 수 있습니다.

import pandas as pd

# Create a sample DataFrame
rpt = pd.DataFrame({
    'STK_ID': ['000002', '600809', '600141', '600329', '603366'],
    'STK_Name': ['Company A', 'Company B', 'Company C', 'Company D', 'Company E'],
    'RPT_Date': ['20120331', '20120331', '20120331', '20120331', '20091231'],
    'sales': [100, 200, 300, 400, 500]
})

# Define the list of stock IDs to filter
stk_list = ['600809', '600141', '600329']

# Filter rows using the `isin` method
filtered_rows = rpt[rpt['STK_ID'].isin(stk_list)]

# Print the filtered DataFrame
print(filtered_rows)
로그인 후 복사

출력:

   STK_ID   STK_Name RPT_Date  sales
0   600809  Company B 20120331     200
1   600141  Company C 20120331     300
2   600329  Company D 20120331     400
로그인 후 복사

isin 메소드는 지정된 열의 값(이 경우 'STK_ID')는 제공된 목록(stk_list)에 포함되어 있습니다. 이 조건을 충족하는 행은 필터링된 DataFrame에 유지됩니다.

위 내용은 값 목록을 기반으로 Pandas DataFrame 행을 필터링하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

원천:php.cn
본 웹사이트의 성명
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.
저자별 최신 기사
인기 튜토리얼
더>
최신 다운로드
더>
웹 효과
웹사이트 소스 코드
웹사이트 자료
프론트엔드 템플릿