如何简化多栏删除？[重复]

Question

我想从数据集中删除多个列。这些列位于随机位置，我有它们的名称。目前，我删除它们如下。

import pandas as pd
data = pd.read_csv('data.cvs')
del data['021']
del data['hg1']
del data['3bm']
del data['og1']
del data['36m']
del data['y7p']
del data['dk4']

有人能告诉我如何简化在更少代码行中删除列的过程吗。

Answer 1

试试这个：

pd.read_csv('data.cvs').drop(['col1','col2'], axis=1)

如果您事先知道所需列的列表，则最好使用usecols参数：

pd.read_csv('data.cvs', usecols=['col_A','col_B'])

在这种情况下，Pandas将不会为不需要的列分配内存，而且它的工作速度也会更快