Bagaimana untuk Menapis Baris DataFrame Pandas Berdasarkan Senarai Nilai?

Menapis Baris DataFrame mengikut Nilai dalam Senarai Set

Dalam panda Python, anda boleh menapis baris berdasarkan sama ada nilai dalam lajur tertentu terdapat dalam set nilai tertentu. Untuk mencapai ini, anda boleh menggunakan kaedah isin, seperti berikut:

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)

Output:

   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

Kaedah isin menyemak sama ada nilai dalam lajur yang ditentukan (dalam kes ini, 'STK_ID') terkandung dalam senarai yang disediakan (stk_list). Baris yang memenuhi syarat ini dikekalkan dalam DataFrame yang ditapis.

Atas ialah kandungan terperinci Bagaimana untuk Menapis Baris DataFrame Pandas Berdasarkan Senarai Nilai?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!