从泡菜文件中读取特定列

Question

所以我知道 Pandas，您可以指定从 csv 文件中提取哪些列以生成 DataFrame

df = pd.read_csv('data.csv', usecols=['a','b','c'])

如何使用腌制文件执行此操作？

df = pd.read_pickle('data.pkl', usecols=['a','b','c'])

给出 TypeError: read_pickle() 得到了一个意外的关键字参数 'usecols'

我在文档中找不到正确的参数

Answer 1

因为 pickle 文件包含完整的 python 对象，我怀疑你可以在加载它们时 select 列，或者至少看起来 pandas 不直接支持它。但是您可以先完全加载它，然后像这样过滤您的列：

df = pd.read_pickle('data.pkl')
df = df.filter(['a', 'b', 'c'])

Documentation