Pandas : Comment convertir rapidement une colonne en liste

Vous pouvez utiliser l’une des méthodes suivantes pour convertir une colonne d’un DataFrame pandas en liste :

Méthode 1 : utilisez tolist()

df['my_column'].tolist()

Méthode 2 : utiliser list()

list(df['my_column'])

Les deux méthodes renverront exactement le même résultat.

Les exemples suivants montrent comment utiliser chacune de ces méthodes avec le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'A', 'A', 'B', 'B', 'B'],
                   'points': [99, 90, 93, 86, 88, 82],
                   'assists': [33, 28, 31, 39, 34, 30]})

#view DataFrame
print(df)

  team  points  assists
0    A      99       33
1    A      90       28
2    A      93       31
3    B      86       39
4    B      88       34
5    B      82       30

Méthode 1 : convertir une colonne en liste à l’aide de tolist()

Le code suivant montre comment utiliser la fonction tolist() pour convertir la colonne « points » du DataFrame en liste :

#convert column to list
my_list = df['points'].tolist()

#view list
print(my_list)

[99, 90, 93, 86, 88, 82]

Nous pouvons confirmer que le résultat est une liste en utilisant la fonction type() :

#check data type
type(my_list)

list

Méthode 2 : convertir une colonne en liste à l’aide de list()

Le code suivant montre comment utiliser la fonction list() pour convertir la colonne « points » du DataFrame en liste :

#convert column to list
my_list = list(df['points'])
#view list
print(my_list)

[99, 90, 93, 86, 88, 82]

Nous pouvons confirmer que le résultat est une liste en utilisant la fonction type() :

#check data type
type(my_list)

list

Notez que les deux méthodes renvoient exactement les mêmes résultats.

Notez que pour les DataFrames extrêmement volumineux, la méthode tolist() a tendance à être la plus rapide.

Ressources additionnelles

Les didacticiels suivants expliquent comment exécuter d’autres fonctions courantes avec les colonnes d’un DataFrame pandas :

Comment supprimer des colonnes dans Pandas
Comment exclure des colonnes dans Pandas
Comment appliquer une fonction aux colonnes sélectionnées dans Pandas
Comment modifier l’ordre des colonnes dans Pandas DataFrame