您能用pycharm查看hdf5文件吗？

Question

是否有一种方式/插件可以查看hdf5文件在py魅力，以便您不必安装HDFVIEW手动检查一个文件？

Answer 1

您可以使用h5py库。

您可能事先不知道HDF5文件的结构。如果是这样的话，您可以使用一个函数来迭代HDF5文件中的所有路径。下面是一个例子：

def traverse_datasets(hdf_file):

    """Traverse all datasets across all groups in HDF5 file."""

    import h5py

    def h5py_dataset_iterator(g, prefix=''):
        for key in g.keys():
            item = g[key]
            path = '{}/{}'.format(prefix, key)
            if isinstance(item, h5py.Dataset): # test for dataset
                yield (path, item)
            elif isinstance(item, h5py.Group): # test for group (go down)
                yield from h5py_dataset_iterator(item, path)

    with h5py.File(hdf_file, 'r') as f:
        for (path, dset) in h5py_dataset_iterator(f):
            print(path, dset)

    return None

示例用法：

traverse_datasets('file.h5')

/DataSet1 <HDF5 dataset "DataSet1": shape (655559, 260), type "<f4">
/DataSet2 <HDF5 dataset "DataSet2": shape (22076, 10000), type "<f4">
/index <HDF5 dataset "index": shape (677635,), type "|V384">

然后，要读取特定的数据集，可以选择一条路径：

with h5pyFile('file.h5', 'r') as f:
    arr = f['/DataSet1'][:]  # read entire dataset into memory

如果无法将数据保存在内存中，则可以迭代打印，也可以将切片提取到内存中。h5py文档有许多例子。语法遵循NumPy约定。