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