Dropping Columns in python pandas

Question

I'm trying to drop all the columns in a pandas dataframe, except for these few, but when I run this code all the columns are dropped. The dataset is so big, that it would be tedious to list them all, any ideas?:

for columns in df:
    if not columns == 'Carbohydrates' or columns == 'Description' or columns == '1st Household Weight' or columns == 'Sugar Total' or columns == 'Kilocalories':
       df = df.drop(columns, axis = 1)

Possible dublicate [How to delete all columns in DataFrame except certain ones?](https://stackoverflow.com/q/45846189/5299969) — Teoretic, Aug 09 '18 at 08:15
droping is not good, you can assign selected columns to variable. like this: `df = df.loc[:, ['a','b']]` — Nihal, Aug 09 '18 at 08:17

score 1 · Answer 1 · edited Aug 09 '18 at 08:34

1

Just select the columns that you want to keep:

df = df[['Carbohydrates','Description','1st Household Weight','Sugar Total','Kilocalories']]

edited Aug 09 '18 at 08:34

Sarvan Kumar

916
10
26

answered Aug 09 '18 at 08:18

Chris Adams

17,620
4
18
35

score 1 · Accepted Answer · answered Aug 09 '18 at 08:18

1

I think you should use the parenthesis in this way:

if not (columns == 'Carbohydrates' or columns == 'Description' or columns == '1st Household Weight' or columns == 'Sugar Total' or columns == 'Kilocalories'):

answered Aug 09 '18 at 08:18

Joe

11,147
5
36
50

Dropping Columns in python pandas

2 Answers2