向h5文件追加数据

最新推荐文章于 2024-06-21 16:39:28 发布

NicoNull

最新推荐文章于 2024-06-21 16:39:28 发布

阅读量470

点赞数

文章标签： python

本文链接：https://blog.csdn.net/qq_33936667/article/details/131480830

版权

该代码示例展示了如何使用h5py库在Python中向H5文件追加数据。首先生成随机特征矩阵，然后定义一个函数write_h5用于将数据写入或追加到H5文件。通过循环将每一层特征写入文件，并用文件名作为数据集的键。最后，打开文件读取并打印所有数据集的内容。

摘要由CSDN通过智能技术生成

向h5文件追加数据

import h5py
import numpy as np
import os

features = np.random.randn(3,3,3)
print(features)
result_path = "train_feature.h5"
def write_h5(feature,name):
    if not os.path.exists(result_path):
        # a 是追加， r是只读， w是写
        file = h5py.File(result_path, "a")
    else:
        file = h5py.File(result_path, 'a')

    file.create_dataset(name=name, data=feature)
    file.close()

for i in range(0,3):
    feat = features[i,:,:]
    name = str(i)
    write_h5(feat, name)

with h5py.File(result_path, 'r') as hf:
    for key in hf.keys():
        print(hf[key].name)  # 获得名称，相当于字典中的key
    for idx in range(0, 3):
        print(hf[str(idx)][:])
    hf.close()