Rumah > pembangunan bahagian belakang > Tutorial Python > Bagaimana untuk Menapis Baris DataFrame Pandas Berdasarkan Senarai Nilai?

Bagaimana untuk Menapis Baris DataFrame Pandas Berdasarkan Senarai Nilai?

Linda Hamilton
Lepaskan: 2024-12-08 20:24:12
asal
421 orang telah melayarinya

How to Filter Pandas DataFrame Rows Based on a List of Values?

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)
Salin selepas log masuk

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
Salin selepas log masuk

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!

sumber:php.cn
Kenyataan Laman Web ini
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Artikel terbaru oleh pengarang
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan