根据列表删除熊猫列

Question

我有一个列表：

my_list = ['a', 'b']

和熊猫数据帧：

d = {'a': [1, 2], 'b': [3, 4], 'c': [1, 2], 'd': [3, 4]}
df = pd.DataFrame(data=d)

我该怎么做才能删除基于列表中的列，在这种情况下，删除列和dfmy_listab

Answer 1

这很简单：df = df.drop(columns=my_list)drop通过指定列名列表来删除列

Answer 2

您也可以选择所需的列：cols_of_interest = ['c', 'd'] df = df[cols_of_interest]如果要删除一系列列：例如 2 到 8，则可以使用：df.drop(df.iloc[:,2:8].head(0).columns, axis=1)

Answer 3

这是一个使用列表理解的简洁脚本：[df.pop（x） 表示my_list中的 x]my_list = ['a', 'b'] d = {'a': [1, 2], 'b': [3, 4], 'c': [1, 2], 'd': [3, 4]} df = pd.DataFrame(data=d)print(df.to_markdown()) |    |   a |   b |   c |   d | |---:|----:|----:|----:|----:| |  0 |   1 |   3 |   1 |   3 | |  1 |   2 |   4 |   2 |   4 | [df.pop(x) for x in my_list]print(df.to_markdown()) |    |   c |   d | |---:|----:|----:| |  0 |   1 |   3 | |  1 |   2 |   4 |