値のリストに基づいて Pandas DataFrame 行をフィルタリングする方法

セット リスト内の値による DataFrame 行のフィルタリング

Python パンダでは、特定の列の値かどうかに基づいて行をフィルタリングできます。は指定された値のセットに存在します。これを実現するには、次のように 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 中国語 Web サイトの他の関連記事を参照してください。