Convertir la colonne Pandas DataFrame en liste

  1. Utiliser la méthode tolist() pour convertir une colonne Dataframe en une liste
  2. Utilisez la fonction list() pour convertir une colonne de Dataframe en liste

Cet article tutoriel présente différentes méthodes pour convertir une colonne Pandas DataFrame en une liste, comme l’utilisation de la méthode tolist() dans Pandas.

Utiliser la méthode tolist() pour convertir une colonne Dataframe en une liste

Une colonne dans DataFrame des Pandas est une Series de Pandas. Donc si nous devons convertir une colonne en liste, nous pouvons utiliser la méthode tolist() dans la Series. La méthode tolist() convertit la Series du DataFrame Pandas en une liste.

Dans le code ci-dessous, df['DOB'] retourne la Series, ou la colonne, avec le nom DOB de la DataFrame.

La méthode tolist() convertit la Series en une liste.

import pandas as pd

df=pd.DataFrame([
        ['James',   '1/1/2014',    '1000'],
        ['Michelina',   '2/1/2014',    '12000'],
        ['Marc',   '3/1/2014',    '36000'],
        ['Bob',   '4/1/2014',    '15000'],
        ['Halena',   '4/1/2014',    '12000']
        ], columns=['Name', 'DOB','Salary'])

print("Pandas DataFrame:\n\n",df,"\n")

list_of_single_column = df['DOB'].tolist()

print("the list of a single column from the dataframe\n",
        list_of_single_column,
        "\n",
        type(list_of_single_column))

Production :

Pandas DataFrame:

         Name       DOB Salary
0      James  1/1/2014   1000
1  Michelina  2/1/2014  12000
2       Marc  3/1/2014  36000
3        Bob  4/1/2014  15000
4     Halena  4/1/2014  12000 

the list of a single column from the dataframe
 ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014'] 
 <class 'list'>

Utilisez la fonction list() pour convertir une colonne de Dataframe en liste

Nous pouvons également utiliser la fonction list() pour convertir une colonne de DataFrame en liste, en passant le DataFrame à la fonction list().

Nous utiliserons les mêmes données que ci-dessus pour démontrer cette approche.

import pandas as pd

df=pd.DataFrame([
        ['James',   '1/1/2014',    '1000'],
        ['Michelina',   '2/1/2014',    '12000'],
        ['Marc',   '3/1/2014',    '36000'],
        ['Bob',   '4/1/2014',    '15000'],
        ['Halena',   '4/1/2014',    '12000']
        ], columns=['Name', 'DOB','Salary'])

print("Pandas DataFrame:\n\n",df,"\n")

list_of_single_column = list(df['DOB'])

print("the list of a single column from the dataframe\n",
        list_of_single_column,
        "\n",
        type(list_of_single_column))

Production :

Pandas DataFrame:

         Name       DOB Salary
0      James  1/1/2014   1000
1  Michelina  2/1/2014  12000
2       Marc  3/1/2014  36000
3        Bob  4/1/2014  15000
4     Halena  4/1/2014  12000 

the list of a single column from the dataframe
 ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014'] 
 <class 'list'>

Article connexe - Pandas DataFrame

  • Comment obtenir une valeur à partir d'une cellule d'un Pandas DataFrame