Can anyone help me to fix this line of code?

Question

I need to use a list comprehension to build a list of features from the columns of data that are not any of 'Name', 'Region', 'state', or 'AdultWeekend'. Can anyone help to fix this?

features = [data.columns for column in data.columns if column not in ['Name', 'Region', 'state', 'Adultweekend']]

Answer 1

Make that column before the for keyword, not data.columns . You want to collect the various values of column over the different loop iterations.

features = [column for column in data.columns if column not in ['Name', 'Region', 'state', 'Adultweekend']]