如何用一行代码删除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 np

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

import pandas pd

df = 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 np

import pandas as pd

input_df.drop(input_df.columns[[np.r_[0,2:4]]],axis=1, inplace = True)