how to get columns pandas code example

Example 1: how to pick out separate columns from the pandas dataframe object

df1 = df.iloc[:, 0:2] # If you want to do it by index. Remember that Python does not slice inclusive of the ending index.
df1 = df[['a', 'b']] ## if you want to do it b nae

Example 2: 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)