python - 将 pandas DataFrame 写入内存缓冲区中的 HDF

Question

我想在内存中获取一个数据帧作为 hdf。下面的代码导致“AttributeError:‘_io.BytesIO’对象没有属性‘put’”。我正在使用 python 3.5 和 pandas 0.17

import pandas as pd
import numpy as np
import io

df = pd.DataFrame(np.arange(8).reshape(-1, 2), columns=['a', 'b'])
buf = io.BytesIO()
df.to_hdf(buf, 'some_key')

更新:正如 UpSampler 指出的那样，“path_or_buf”不能是 io 流(我觉得这很困惑，因为 buf 通常可以是 io 流，请参阅 to_csv )。除了写入磁盘并将其读回之外，我能否在内存中获取数据帧作为 hdf？

Answer 1

你的第一个参数 df.to_hdf()必须是“路径(字符串)或 HDFStore 对象”而不是 io 流。文档:http://pandas.pydata.org/pandas-docs/version/0.17.0/generated/pandas.DataFrame.to_hdf.html