如何使用python将Hdf5文件部分复制到保持相同结构的新文件?

时间:2014-07-01 12:25:16

标签: python hdf5 h5py

我有一个大的hdf5文件,看起来像这样:

A/B/dataset1, dataset2
A/C/dataset1, dataset2
A/D/dataset1, dataset2
A/E/dataset1, dataset2

...

我想创建一个仅包含以下内容的新文件: A / B / dataset1,dataset2 A / C / dataset1,dataset2

python中最简单的方法是什么?

我做了:

fs = h5py.File('source.h5', 'r')
fd = h5py.File('dest.h5', 'w')
fs.copy('group B', fd)

问题是我得到的是dest.h5:

B/dataset1, dataset2

并且我遗漏了一部分树枝。

1 个答案:

答案 0 :(得分:23)

fs.copy('A/B', fd)不会将路径/A/B/复制到fd,它只会复制群组B(正如您发现的那样!)。所以你首先需要创建路径的其余部分:

fd.create_group('A')
fs.copy('A/B', fd['/A'])

或者,如果你将经常使用这个小组:

fd_A = fd.create_group('A')
fs.copy('A/B', fd_A)

这会将Bfs['/A/B']复制到fd['/A']

In [1]: fd['A/B'].keys()
Out[1]: [u'dataset1', u'dataset2']

这是一种自动执行此操作的方法:

# Get the name of the parent for the group we want to copy
group_path = fs['/A/B'].parent.name

# Check that this group exists in the destination file; if it doesn't, create it
# This will create the parents too, if they don't exist
group_id = fd.require_group(group_path)

# Copy fs:/A/B/ to fd:/A/G
fs.copy('/A/B', group_id, name="G")

print(fd['/A/G'].keys())
# [u'dataset1', u'dataset2']