python - 确定 pandas HDF 文件中 DataFrame 的格式

Question

有一个 HDF 文件“file.h5”，其中保存的 pandas DataFrame(或系列)的键名是“df”。如何确定“df”以何种格式(即“固定”或“表格”)保存到文件中？

感谢您的帮助!

Answer 1

有点晚了，但也许其他人会发现它有帮助。

您可以解析 HDFStore.info() 的输出. table 格式的对象具有 appendable 类型:

>>> print(h5_table.info())
<class 'pandas.io.pytables.HDFStore'>
File path: /tmp/df_table.h5
/df            frame_table  (typ->appendable,nrows->2,ncols->2,indexers->[index],dc->[])

>>> print(h5_fixed.info())
<class 'pandas.io.pytables.HDFStore'>
File path: /tmp/df_fixed.h5
/df            frame        (shape->[2,2]) 

这是一个最小的(即没有丢失文件或 key 的错误处理)示例:

def get_hd5_format(path, key):
    with pd.HDFStore(path) as store:
        info = store.info()
    return 'table' if 'typ->appendable' in next(k for k in info.splitlines()[2:] if k.startswith('/'+key)).split()[2] else 'fixed'

示例用法:

>>> get_hd5_format('/tmp/df_table.h5', 'df')
'table'
>>> get_hd5_format('/tmp/df_fixed.h5', 'df')
'fixed'