如果您事先知道列名,则可以通过设置
usecols参数来实现
当您知道要使用哪些列时
假设您有带有列的csv文件,
['id','name','last_name']而您只想要
['name','last_name']。您可以按照以下步骤进行操作:
import pandas as pddf = pd.read_csv("sample.csv", usecols = ['name','last_name'])
当您想要前N列时
如果您不知道列名,但想要从数据框中获取前N列。你可以做到
import pandas as pddf = pd.read_csv("sample.csv", usecols = [i for i in range(n)])
编辑
当您知道要删除的列的名称时
# Read column names from filecols = list(pd.read_csv("sample_data.csv", nrows =1))print(cols)# Use list comprehension to remove the unwanted column in **usecol**df= pd.read_csv("sample_data.csv", usecols =[i for i in cols if i != 'name'])