/Python/How to get a dataframe column as a list

How to get a dataframe column as a list

import pandas as pd

data_dict = {'one': pd.Series([1, 2, 3], index=['a', 'b', 'c']),
             'two': pd.Series([1, 2, 3, 4], index=['a', 'b', 'c', 'd'])}

df = pd.DataFrame(data_dict)

print(f"DataFrame:\n{df}\n")
print(f"column types:\n{df.dtypes}")

col_one_list = df['one'].tolist()

col_one_arr = df['one'].to_numpy()

print(f"\ncol_one_list:\n{col_one_list}\ntype:{type(col_one_list)}")
print(f"\ncol_one_arr:\n{col_one_arr}\ntype:{type(col_one_arr)}")

Source: stackoverflow.com

Similar codes

31 votes108 views5 answers
How to place image in tkinter
Python
0 vote103 views1 answer
Py regex if .jpg
Python
25 votes134 views6 answers
Python check if list contains elements of another list
Python
6 votes124 views3 answers
From sklearn.metrics import mean_square_error
Python
0 vote104 views1 answer
Python disable logging on unittest
Python
81 votes113 views6 answers
String to date python
Python
3 votes105 views1 answer
Restore index after concatenate
Python
31 votes110 views6 answers
Upgrade python version
Python
1 vote106 views1 answer
Python inheritance remove an attribute
Python
1 vote104 views2 answers
Number of digits calculation
Python