从泡菜文件中读取特定列

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对象，所以我怀疑您可以在加载它们时选择列，或者至少看起来pandas不直接支持它。但是您可以先完全加载它，然后按以下方式过滤您的列：

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

Documentation