Pandas : comment sélectionner des colonnes en fonction de la condition

Vous pouvez utiliser les méthodes suivantes pour sélectionner des colonnes dans un DataFrame pandas par condition :

Méthode 1 : sélectionner les colonnes dont au moins une ligne remplit la condition

#select columns where at least one row has a value greater than 2
df.loc[:, (df > 2).any()]

Méthode 2 : sélectionner les colonnes où toutes les lignes remplissent la condition

#select columns where all rows have a value greater than 2
df.loc[:, (df > 2).all()] 

Méthode 3 : sélectionner les colonnes dont au moins une ligne répond à plusieurs conditions

#select columns where at least one row has a value between 10 and 15
df.loc[:, ((df>=10) & (df<=15)).any()]

Les exemples suivants montrent comment utiliser chaque méthode avec le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'apples': [7, 3, 3, 4, 3],
                   'oranges': [2, 0, 2, 0, 1],
                   'bananas': [5, 0, 4, 0, 12]},
                    index=['Farm1', 'Farm2', 'Farm3', 'Farm4', 'Farm5'])

#view DataFrame
print(df)

       apples  oranges  bananas
Farm1       7        2        5
Farm2       3        0        0
Farm3       3        2        4
Farm4       4        0        0
Farm5       3        1       12

Exemple 1 : sélectionner les colonnes dont au moins une ligne remplit la condition

Nous pouvons utiliser le code suivant pour sélectionner les colonnes du DataFrame où au moins une ligne de la colonne a une valeur supérieure à 2 :

#select columns where at least one row has a value greater than 2
df.loc[:, (df > 2).any()]

	apples	bananas
Farm1	7	5
Farm2	3	0
Farm3	3	4
Farm4	0	0
Farm5	3	12

Notez que les colonnes pommes et bananes sont renvoyées car ces deux colonnes ont au moins une ligne avec une valeur supérieure à 2.

Exemple 2 : sélectionner les colonnes où toutes les lignes remplissent la condition

Nous pouvons utiliser le code suivant pour sélectionner les colonnes du DataFrame où chaque ligne de la colonne a une valeur supérieure à 2 :

#select columns where every row has a value greater than 2
df.loc[:, (df > 2).all()]

	apples
Farm1	7
Farm2	3
Farm3	3
Farm4	4
Farm5	3

Notez que seule la colonne des pommes est renvoyée car c’est la seule colonne où chaque ligne de la colonne a une valeur supérieure à 2.

Exemple 3 : sélectionner les colonnes dont au moins une ligne remplit plusieurs conditions

Nous pouvons utiliser le code suivant pour sélectionner les colonnes du DataFrame où au moins une ligne de la colonne a une valeur comprise entre 10 et 15 :

#select columns where every row has a value greater than 2
df.loc[:, ((df>=10) & (df<=15)).any()]

	bananas
Farm1	5
Farm2	0
Farm3	4
Farm4	0
Farm5	12

Notez que seule la colonne bananes est renvoyée car c’est la seule colonne où au moins une ligne de la colonne a une valeur comprise entre 10 et 15.

Ressources additionnelles

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

Comment sélectionner des colonnes par nom dans Pandas
Comment sélectionner des colonnes par index dans Pandas
Comment sélectionner des colonnes contenant une chaîne spécifique dans Pandas