Pandas : rechercher une chaîne dans toutes les colonnes de DataFrame

Vous pouvez utiliser la syntaxe suivante pour rechercher une chaîne particulière dans chaque colonne d’un DataFrame pandas et filtrer les lignes contenant la chaîne dans au moins une colonne :

#define filter
mask = np.column_stack([df[col].str.contains(r"my_string", na=False) for col in df])

#filter for rows where any column contains 'my_string'
df.loc[mask.any(axis=1)]

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : rechercher une chaîne dans toutes les colonnes de Pandas DataFrame

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur le premier rôle et le deuxième rôle de divers joueurs de basket-ball dans une équipe :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'player': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   'first_role': ['P Guard', 'P Guard', 'S Guard', 'S Forward',
                                  'P Forward', 'Center', 'Center', 'Center'],
                   'second_role': ['S Guard', 'S Guard', 'Forward', 'S Guard',
                                   'S Guard', 'S Forward', 'P Forward', 'P Forward']})

#view DataFrame
print(df)

  player first_role second_role
0      A    P Guard     S Guard
1      B    P Guard     S Guard
2      C    S Guard     Forward
3      D  S Forward     S Guard
4      E  P Forward     S Guard
5      F     Center   S Forward
6      G     Center   P Forward
7      H     Center   P Forward

Le code suivant montre comment filtrer le DataFrame pandas pour les lignes où la chaîne « Guard » apparaît dans n’importe quelle colonne :

import numpy as np

#define filter
mask = np.column_stack([df[col].str.contains(r"Guard", na=False) for col in df])

#filter for rows where any column contains 'Guard'
df.loc[mask.any(axis=1)]

        player	first_role  second_role
0	A	P Guard	    S Guard
1	B	P Guard	    S Guard
2	C	S Guard	    Forward
3	D	S Forward   S Guard
4	E	P Forward   S Guard

Notez que chaque ligne du DataFrame résultant contient la chaîne « Guard » dans au moins une colonne.

Vous pouvez également filtrer les lignes où une chaîne parmi plusieurs apparaît dans au moins une colonne en utilisant l’opérateur « OU » ( | ) dans les pandas.

Par exemple, le code suivant montre comment filtrer les lignes où « P Guard » ou « Center » apparaît dans au moins une colonne :

import numpy as np

#define filter
mask = np.column_stack([df[col].str.contains(r"P Guard|Center", na=False) for col in df])

#filter for rows where any column contains 'P Guard' or 'Center'
df.loc[mask.any(axis=1)]

        player	first_role  second_role
0	A	P Guard	    S Guard
1	B	P Guard	    S Guard
5	F	Center	    S Forward
6	G	Center	    P Forward
7	H	Center	    P Forward

Notez que chaque ligne du DataFrame résultant contient « P Guard » ou « Center » dans au moins une colonne.

Remarque : Il est important d’inclure l’argument na=False dans la fonction contain() sinon vous rencontrerez une erreur si des valeurs NaN sont présentes dans le DataFrame.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations de filtrage courantes dans les pandas :

Comment filtrer un DataFrame Pandas par valeurs de colonne
Comment filtrer les lignes Pandas DataFrame par date
Comment filtrer un DataFrame Pandas sur plusieurs conditions