根据列表删除熊猫列

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 |