Comment sélectionner les lignes Pandas DataFrame en fonction d'une liste de valeurs ?

Sélection de lignes à partir d'un DataFrame basée sur une liste de valeurs dans Pandas

Cette question aborde le défi du sous-ensemble de lignes à partir d'un dataframe Pandas basé sur une liste de valeurs.

Question :

Étant donné le cadre de données :

df = DataFrame({'A': [5, 6, 3, 4], 'B': [1, 2, 3, 5]})

Comment pouvons-nous sélectionner les lignes où les valeurs de la colonne « A » correspondent aux éléments dans une liste, telle comme :

list_of_values = [3, 6]

Réponse :

Pour filtrer la trame de données en fonction d'une liste de valeurs, nous pouvons utiliser la méthode isin :

y = df[df['A'].isin(list_of_values)]

Résultat :

   A  B
1  6  2
2  3  3

Pour exclure les lignes dont les valeurs ne sont pas présentes dans la liste, on peut utiliser l'opérateur non logique (~) avec isin :

z = df[~df['A'].isin(list_of_values)]

Résultat :

   A  B
0  5  1
3  4  5

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!