how to select columns in pandas with names code example
Example 1: python select columns names from dataframe
# Import pandas package
import pandas as pd
# making data frame
data = pd.read_csv("nba.csv")
# iterating the columns
for col in data.columns:
print(col)
Example 2: select columns pandas
df1 = df.iloc[:,0:2] # Remember that Python does not slice inclusive of the ending index.
Example 3: select specific column names from dataframe
import pandas
data = pandas.DataFrame({'A' : ['X', 'Y'],
'B' : 1,
'C' : [2, 3]})
print data[['A', 'B']]
# Output A B
# 0 X 1
# 1 Y 1