Converter a coluna Pandas DataFrame em lista
-
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'>