首頁 > 後端開發 > Python教學 > 如何根據值列表過濾 Pandas DataFrame 行?

如何根據值列表過濾 Pandas DataFrame 行?

Linda Hamilton
發布: 2024-12-08 20:24:12
原創
446 人瀏覽過

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
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板