Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Drop multiple columns in pandas

Tags:

python

pandas

I am trying to drop multiple columns (column 2 and 70 in my data set, indexed as 1 and 69 respectively) by index number in a pandas data frame with the following code:

df.drop([df.columns[[1, 69]]], axis=1, inplace=True)

I get the following error:

TypeError: unhashable type: 'Index'

And in my code the [1, 69] is highlighted and says:

Expected type 'Integral', got 'list[int]' instead

The following code does what I want it to do successfully, but on two lines of repetitive code (first dropping col index 69, then 1, and order does matter because dropping earlier columns changes the index of later columns). I thought I could specify more than one column index simply as a list, but perhaps I have something wrong above?

df.drop([df.columns[69]], axis=1, inplace=True)
df.drop([df.columns[1]], axis=1, inplace=True)

Is there a way that I can do this on one line similar to the first code snippet above?

like image 551
lukewitmer Avatar asked Oct 13 '14 19:10

lukewitmer


2 Answers

You don't need to wrap it in a list with [..], just provide the subselection of the columns index:

df.drop(df.columns[[1, 69]], axis=1, inplace=True)

as the index object is already regarded as list-like.

like image 142
joris Avatar answered Nov 13 '22 23:11

joris


Try this

df.drop(df.iloc[:, 1:69], inplace=True, axis=1)

This works for me

like image 29
Okroshiashvili Avatar answered Nov 14 '22 01:11

Okroshiashvili