Pandas : Comment extraire des lignes de DataFrame

Vous pouvez utiliser la fonction pop() pour supprimer rapidement une colonne d’un DataFrame pandas.

Afin d’utiliser la fonction pop() pour supprimer des lignes, vous devez d’abord transposer le DataFrame puis utiliser la fonction pop() pour supprimer les colonnes (c’est-à-dire les lignes du DataFrame d’origine) :

#pop the row in index position 3
df.T.pop(3)

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

Exemple : afficher les lignes du DataFrame pandas

Supposons que nous ayons le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'B', 'C', 'D', 'E', 'F'],
                   'points': [18, 22, 19, 14, 14, 11],
                   'assists': [5, 7, 7, 9, 12, 9]})

#view DataFrame
print(df)

  team  points  assists
0    A      18        5
1    B      22        7
2    C      19        7
3    D      14        9
4    E      14       12
5    F      11        9

Supposons maintenant que nous souhaitions supprimer la ligne en position d’index 3 du DataFrame.

Nous pouvons transposer le DataFrame puis utiliser la fonction pop() pour supprimer la ligne en position d’index 3 :

#define transposed DataFrame
df_transpose = df.T

#remove row in index position 3 of original DataFrame
df_transpose.pop(3)

team        D
points     14
assists     9
Name: 3, dtype: object

Nous pouvons ensuite transposer à nouveau le DataFrame pour récupérer le DataFrame d’origine avec une ligne supprimée :

#transpose back to original DataFrame
df = df_transpose.T

#view updated DataFrame
print(df)

  team points assists
0    A     18       5
1    B     22       7
2    C     19       7
4    E     14      12
5    F     11       9

Notez que la ligne en position d’index 3 a été supprimée du DataFrame.

Toutes les autres lignes du DataFrame restent intactes.

Remarque : Vous pouvez trouver la documentation complète de la fonction pop() dans pandas ici .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans les pandas :

Comment insérer une ligne dans un DataFrame Pandas
Comment supprimer la première ligne dans Pandas DataFrame
Comment supprimer des lignes dans Pandas DataFrame en fonction de la condition