Converter a coluna Pandas DataFrame em lista

Criado: February-06, 2021

Utilize o método tolist() para converter uma coluna DataFrame para uma lista
Use a função list() para converter uma coluna de dataframe em uma lista

Este artigo tutorial irá introduzir diferentes métodos para converter uma coluna Pandas DataFrame para uma lista, como a utilização do método tolist() em Pandas.

Utilize o método `tolist()` para converter uma coluna DataFrame para uma lista

Uma coluna no dataframe Pandas é um Pandas Series. Assim, se precisarmos de converter uma coluna numa lista, podemos utilizar o método tolist() na Series. O método tolist() converte a Series de pandas data-frame para uma lista.

No código abaixo, df['DOB'] devolve a Series, ou a coluna, com o nome como DOB da DataFrame.

O método tolist() converte a Series para uma lista.

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))

Resultado:

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'>

Use a função `list()` para converter uma coluna de dataframe em uma lista

Podemos também utilizar a função list() para converter uma coluna DataFrame para uma lista, passando a DataFrame para a função list().

Utilizaremos os mesmos dados que os acima referidos para demonstrar esta abordagem.

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))

Resultado:

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'>