根据列名在两列之间删除熊猫数据框中的多列

Question

A super simple question, for which I cannot find an answer.

I have a dataframe with 1000+ columns and cannot drop by column number, I do not know them. I want to drop all columns between two columns, based on their names.

foo = foo.drop(columns = ['columnWhatever233':'columnWhatever826']) 

does not work. I tried several other options, but do not see a simple solution. Thanks!

Answer 1

You can use .loc with column range. For example if you have this dataframe:

   A  B  C  D  E
0  1  3  3  6  0
1  2  2  4  9  1
2  3  1  5  8  4

Then to delete columns B to D :

df = df.drop(columns=df.loc[:, "B":"D"].columns)
print(df)

Prints:

   A  E
0  1  0
1  2  1
2  3  4