如何简化多列删除？

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不会为不需要的列分配内存，而且工作速度也会更快