return selected column names of a datafrmae code example
Example 1: pd dataframe get column names
lst = data.columns.values # data is dataframe
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)