Maison > développement back-end > Tutoriel Python > Comment filtrer les lignes Pandas DataFrame en fonction d'une liste de valeurs ?

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

Linda Hamilton
Libérer: 2024-12-08 20:24:12
original
421 Les gens l'ont consulté

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

Filtrer les lignes du DataFrame par valeurs dans une liste définie

Dans les pandas Python, vous pouvez filtrer les lignes selon que la valeur dans une colonne spécifique est présent dans un ensemble de valeurs donné. Pour y parvenir, vous pouvez utiliser la méthode isin, comme suit :

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)
Copier après la connexion

Sortie :

   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
Copier après la connexion

La méthode isin vérifie si les valeurs de la colonne spécifiée (dans ce cas, 'STK_ID') sont contenus dans la liste fournie (stk_list). Les lignes qui remplissent cette condition sont conservées dans le DataFrame filtré.

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!

source:php.cn
Déclaration de ce site Web
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn
Derniers articles par auteur
Tutoriels populaires
Plus>
Derniers téléchargements
Plus>
effets Web
Code source du site Web
Matériel du site Web
Modèle frontal