如何在使用pandas读取csv文件时删除该文件的特定列？

Question

在使用pandas加载csv时，我需要删除带有labelname的列。我正在阅读csv如下，并想在它里面添加参数这样做。谢谢。

pd.read_csv("sample.csv")

我知道在阅读csv之后要做的事情：

df.drop('name', axis=1)

Answer 1

使用pd.read_csv和nrows=1从CSV中获取列标题，然后使用usecols执行后续读取，以提取除了要忽略的列之外的所有内容。

headers = [*pd.read_csv('sample.csv', nrows=1)]
df = pd.read_csv('sample.csv', usecols=[c for c in headers if c != 'name']))

或者，您也可以使用CSV模块执行同样的操作（只读取标题）非常有效

import csv

with open("sample.csv", 'r') as f:
    header = next(csv.reader(f))
    # For python 2, use
    # header = csv.reader(f).next()

df = pd.read_csv('sample.csv', usecols=list(set(header) - {'name'}))