Pandas: Get a list of a specified column of a DataFrame

Last update on December 21 2024 07:42:07 (UTC/GMT +8 hours)

Write a Pandas program to get a list of a specified column of a DataFrame.
Sample data:
Powered by
Original DataFrame
col1 col2 col3
0 1 4 7
1 2 5 8
2 3 6 9
Col2 of the DataFrame to list:
[4, 5, 6]

Sample Solution :-

Python Code :

import pandas as pd
d = {'col1': [1, 2, 3], 'col2': [4, 5, 6], 'col3': [7, 8, 9]}
df = pd.DataFrame(data=d)
print("Original DataFrame")
print(df)
col2_list = df["col2"].tolist()
print("Col2 of the DataFrame to list:")
print(col2_list)

Sample Output:

 Powered by 
Original DataFrame
   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6     9
Col2 of the DataFrame to list:
[4, 5, 6]                 

Explanation:

The above code creates a dictionary ‘d’ containing 3 columns with some sample data, then uses the pd.DataFrame() function from pandas to create a DataFrame ‘df’ from this dictionary.

col2_list = df["col2"].tolist(): This line of code extracts the column named col2 from the DataFrame df using df["col2"], and converts it into a Python list using the tolist() method. The resulting col2_list variable contains the values of the col2 column in a list format.

Python-Pandas Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Pandas program to rename a specific column name in a given DataFrame.
Next: Write a Pandas program to create a DataFrame from a Numpy array and specify the index column and column headers.