Wie filtere ich Pandas DataFrame-Zeilen basierend auf einer Liste von Werten in einer bestimmten Spalte?

DataFrame-Zeilen filtern, wenn der Wert in der Spalte mit einer bestimmten Liste übereinstimmt

Das Pandas DataFrame-RPT verfügt über mehrere Zeilen mit Daten zu Beständen. Um Zeilen abzurufen, in denen die Aktien-IDs mit einer bestimmten Werteliste übereinstimmen, sollte die isin-Methode verwendet werden. Sehen wir uns an, wie man DataFrame-Zeilen mithilfe einer Werteliste filtert:

import pandas as pd

# Create a sample DataFrame
rpt = pd.DataFrame({'STK_ID': ['000002', '600809', '600141', '600329', '603366'],
                   'RPT_Date': ['20120331', '20120331', '20120331', '20120331', '20091231'],
                   'sales': [100, 200, 300, 400, 500]})

# Stock IDs to filter for
stk_list = ['600809', '600141', '600329']

# Filter rows using isin
filtered_rows = rpt[rpt['STK_ID'].isin(stk_list)]

Die isin-Methode prüft, ob die Werte in der angegebenen Spalte in der bereitgestellten Liste vorhanden sind. Zeilen mit übereinstimmenden IDs werden in den DataFrame filtered_rows gefiltert:

print(filtered_rows)

   STK_ID  RPT_Date  sales
0  600809  20120331    200
1  600141  20120331    300
2  600329  20120331    400

Dieser Filteransatz vermeidet den häufigen Fehler der Verwendung von in, der für die Überprüfung, ob Werte in einer Liste vorhanden sind, nicht unterstützt wird. Durch die Verwendung von isin können die gewünschten Zeilen effizient aus dem DataFrame abgerufen werden.

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