python dataframe pandas drop column using int

Question

I understand that to drop a column you use df.drop('column name', axis=1). Is there a way to drop a column using a numerical index instead of the column name?

Answer 1

 If you want to drop multiple columns like this:

cols = [1,2,4,5,12]

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

inplace=True is used so that it can make the changes in the dataframe itself without doing the column dropping on a copy of the data frame. If you need to keep your original intact, refer the following code:

df_after_dropping = df.drop(df.columns[cols],axis=1)

If you do not work with inplace=True then you will have to do df = df.drop() 

If you are interested in learning Python and want to become a certified Python expert, then check out Python Courses and yourself.upskill