Answers for "pandas dataframe get column list"

5

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)}")
Posted by: Guest on June-09-2020
1

dataframe column in list

In [1]: df = pd.DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})

In [2]: df
Out[2]:
   A  B
0  5  1
1  6  2
2  3  3
3  4  5

In [3]: df[df['A'].isin([3, 6])]
Out[3]:
   A  B
1  6  2
2  3  3
Posted by: Guest on September-28-2020

Code answers related to "pandas dataframe get column list"

Python Answers by Framework

Browse Popular Code Answers by Language