如何用一行代码删除python中的多列？

Question

我正在尝试删除数据框中的以下列：1,2,101:117,121:124,126。

到目前为止，我发现删除列的两种方法是：

df.drop(df.columns[2:6],axis=1)

df.drop(df.columns[[0,3,5]],axis=1) 

但是如果我尝试

df.drop(df.columns[1,2,101:117,121:124],axis=1)

我收到“索引过多”错误

我也试过这个

a=df.drop(df.columns[[1,2]],axis=1)

b=a.drop(a.columns[99:115],axis=1)

c=b.drop(b.columns[102:105],axis=1)

d=c.drop(c.columns[103],axis=1)

但这并不是出于某种原因删除我想要的列。

Answer 1

使用np.r_到切片：import numpy as npdf.drop(columns=df.columns[np.r_[1, 2, 101:117, 121:124, 126]])import pandas pddf = pd.DataFrame(np.random.randint(1, 10, (2, 130)))df.drop(columns=df.columns[np.r_[1, 2, 101:117, 121:124, 126]])#     0    3    4    5    6 ... 120 124 125  127#0    6    1    3    7    2 ...   8   7   2    6#1    1    9    2    5    3 ...   7   3   9    4

Answer 2

这应该有效：df.drop(df.columns[[indexes_of_columns_you_want_to_delete]],axis=1,inplace=True)

Answer 3

请试试这个：import numpy as npimport pandas as pdinput_df.drop(input_df.columns[[np.r_[0,2:4]]],axis=1, inplace = True)