Comment filtrer les lignes Pandas DataFrame en fonction d'une liste de valeurs dans une colonne spécifique ?

Filtrer les lignes du DataFrame si la valeur dans la colonne correspond à une liste spécifique

Le rpt DataFrame pandas contient plusieurs lignes de données liées aux actions. Pour récupérer les lignes où les identifiants de stock correspondent à une liste spécifique de valeurs, la méthode isin doit être utilisée. Voyons comment filtrer les lignes DataFrame à l'aide d'une liste de valeurs :

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)]

La méthode isin vérifie si les valeurs de la colonne spécifiée sont présentes dans la liste fournie. Les lignes avec des ID correspondants sont filtrées dans le DataFrame filtered_rows :

print(filtered_rows)

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

Cette approche de filtrage évite l'erreur courante de l'utilisation de in, qui n'est pas prise en charge pour vérifier si les valeurs sont dans une liste. En utilisant isin, les lignes souhaitées peuvent être récupérées efficacement à partir du DataFrame.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!