Wie filtere ich Pandas DataFrame-Zeilen basierend auf einer Werteliste?

DataFrame-Zeilen nach Werten in einer Satzliste filtern

In Python-Pandas können Sie Zeilen basierend darauf filtern, ob der Wert in einer bestimmten Spalte vorhanden ist ist in einer bestimmten Menge von Werten vorhanden. Um dies zu erreichen, können Sie die isin-Methode wie folgt verwenden:

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)

Ausgabe:

   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

Die isin-Methode prüft, ob die Werte in der angegebenen Spalte (in diesem Fall 'STK_ID') sind in der bereitgestellten Liste (stk_list) enthalten. Zeilen, die diese Bedingung erfüllen, werden im gefilterten DataFrame beibehalten.

Das obige ist der detaillierte Inhalt vonWie filtere ich Pandas DataFrame-Zeilen basierend auf einer Werteliste?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!