Convert Pandas DataFrame Column to List
Usama Imtiaz
Dec-10, 2020
Dec-05, 2020
Pandas
Pandas DataFrame Column
Pandas DataFrame
-
Use the
tolist()Method to Convert a Dataframe Column to a List -
Use the
list()Function to Convert a Dataframe Column to a List
This tutorial article will introduce different methods to convert a Pandas DataFrame column to a list, like using the tolist() method in Pandas.
Use the tolist() Method to Convert a Dataframe Column to a List
A column in the Pandas dataframe is a Pandas Series. So if we need to convert a column to a list, we can use the tolist() method in the Series. tolist() converts the Series of pandas data-frame to a list.
In the code below, df['DOB'] returns the Series, or the column, with the name as DOB from the DataFrame.
The tolist() method converts the Series to a list.
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))
Output:
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 the list() Function to Convert a Dataframe Column to a List
We can also use the list() function to convert a DataFrame column to a list, by passing the DataFrame to the list() function.
We will use the same data as above to demonstrate this approach.
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))
Output:
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'>
