Comment obtenir la première colonne de Pandas DataFrame (avec exemples)

Vous pouvez utiliser la syntaxe suivante pour obtenir la première colonne d’un DataFrame pandas :

df.iloc[:, 0]

Le résultat est une série de pandas. Si vous souhaitez que le résultat soit un DataFrame pandas, vous pouvez utiliser la syntaxe suivante :

df.iloc[:, :1]

Les exemples suivants montrent comment utiliser cette syntaxe dans la pratique.

Exemple 1 : obtenir la première colonne de Pandas DataFrame (retourner une série)

Le code suivant montre comment obtenir la première colonne d’un DataFrame pandas :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19, 23, 25, 29],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4],
                   'rebounds': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print(df)

   points  assists  rebounds
0      25        5        11
1      12        7         8
2      15        7        10
3      14        9         6
4      19       12         6
5      23        9         5
6      25        9         9
7      29        4        12

#get first column
first_col = df.iloc[:, 0]

#view first column
print(first_col)

0    25
1    12
2    15
3    14
4    19
5    23
6    25
7    29
Name: points, dtype: int64

Le résultat est une série de pandas :

#check type of first_col
print(type(first_col))

<class 'pandas.core.series.Series'>

Exemple 2 : obtenir la première colonne de Pandas DataFrame (retourner un DataFrame)

Le code suivant montre comment obtenir la première colonne d’un DataFrame pandas et renvoyer un DataFrame en conséquence :

#get first column (and return a DataFrame)
first_col = df.iloc[:, :1]

#view first column
print(first_col)

   points
0      25
1      12
2      15
3      14
4      19
5      23
6      25
7      29

#check type of first_col
print(type(first_col))

<class 'pandas.core.frame.DataFrame'>

Ressources additionnelles

Comment ajouter une colonne à un DataFrame Pandas
Comment insérer une colonne dans un DataFrame Pandas
Comment créer une nouvelle colonne basée sur une condition dans Pandas