세트 목록의 값으로 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 중국어 웹사이트의 기타 관련 기사를 참조하세요!