按集合列表中的值过滤 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中文网其他相关文章！
