Feature = df_adv_data[['Newspaper','Radio','TV']] code example Example: Feature = df_adv_data[['Newspaper','Radio','TV']] Feature = df_adv_data[['Newspaper','Radio','TV']]