从泡菜文件中读取特定列

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() got an unexpected keyword argument 'usecols'

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

Answer 1

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

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

Documentation