Comment puis-je filtrer les lignes Pandas DataFrame en fonction d'une liste de valeurs ?

Filtrage des lignes de dataframe en fonction des valeurs d'une liste

Dans le contexte des dataframes Python, le filtrage des lignes en fonction de valeurs spécifiques peut être réalisé en utilisant la bibliothèque des pandas. Cette question répond à l'obligation de filtrer les lignes où la valeur d'une colonne particulière correspond à l'un des éléments d'une liste fournie.

Pour ce faire, la méthode isin() doit être utilisée, par opposition à la méthode tentée dans opérateur qui n’est pas reconnu par les pandas. L'expression rpt['STK_ID'].isin(stk_list) vérifie si les valeurs de la colonne STK_ID sont présentes dans la variable stk_list, sélectionnant ainsi les lignes qui remplissent cette condition.

Par exemple, considérons un dataframe rpt :

rpt
<class 'pandas.core.frame.DataFrame'>
MultiIndex: 47518 entries, ('000002', '20120331') to ('603366', '20091231')
Data columns:
STK_ID                    47518  non-null values
STK_Name                  47518  non-null values
RPT_Date                  47518  non-null values
sales                     47518  non-null values

Pour filtrer les lignes où la valeur STK_ID est soit '600809', '600141' ou '600329' :

stk_list = ['600809', '600141', '600329']

rst = rpt[rpt['STK_ID'].isin(stk_list)]

La première trame de données contiendra désormais uniquement les lignes correspondant aux identifiants de stock spécifiés dans la stk_list.

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!