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

열의 값이 특정 목록과 일치하면 DataFrame 행 필터링

pandas DataFrame rpt에는 주식과 관련된 여러 데이터 행이 있습니다. 주식 ID가 특정 값 목록과 일치하는 행을 검색하려면 isin 메소드를 사용해야 합니다. 값 목록을 사용하여 DataFrame 행을 필터링하는 방법을 살펴보겠습니다.

import pandas as pd

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

# Stock IDs to filter for
stk_list = ['600809', '600141', '600329']

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

isin 메서드는 지정된 열의 값이 제공된 목록에 있는지 확인합니다. 일치하는 ID가 있는 행은 Filtered_rows DataFrame으로 필터링됩니다.

print(filtered_rows)

   STK_ID  RPT_Date  sales
0  600809  20120331    200
1  600141  20120331    300
2  600329  20120331    400

이 필터링 접근 방식은 값이 목록에 있는지 확인하는 데 지원되지 않는 in 사용의 일반적인 오류를 방지합니다. isin을 활용하면 DataFrame에서 원하는 행을 효율적으로 검색할 수 있습니다.

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