w3resource

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


43. Column to List

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.


For more Practice: Solve these Related Problems:

  • Write a Pandas program to extract a specified column from a DataFrame and convert it into a Python list, then sort the list.
  • Write a Pandas program to get a list of values from a column and then remove duplicates from the list.
  • Write a Pandas program to retrieve a column as a list and then filter the list for values above a given threshold.
  • Write a Pandas program to extract a column into a list and then compute the frequency of each unique element in that list.

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.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.