Gets DataFrame column names of the three methods

df= pd.DataFrame({'a': range(10, 20), 'b': range(20, 30)})
df

1. list down style

[column for column in df]
[a,b]

2. columns property

columns property returns Index, columns.values property returns numpy.ndarray, may then be tolist (), or a list (ndarray) is converted to list

print(type(df.columns))
<class 'pandas.core.indexes.base.Index'>
print(type(df.columns.values))
<class 'numpy.ndarray'>
print(type(df.columns.tolist()),":"+"\n",df.columns.tolist())
<class 'list'> :
 ['a', 'b']
print(type(df.columns.values.tolist()),":"+"\n",df.columns.values.tolist())
<class 'list'> :
 ['a', 'b']

 

3. Direct use list, returns a list containing a listing of columnsd

print(list(df))
['a', 'b']

 

Guess you like

Origin www.cnblogs.com/wqbin/p/11845042.html