如何使用h5py读入数据

Posted 2021-11-30 hellojiaojiao

在网上找到的方法，如下：

import h5py
import numpy as np
#HDF5的写入：
imgData = np.zeros((2,4))
f = h5py.File(‘HDF5_FILE.h5‘,‘w‘) #创建一个h5文件，文件指针是f
f[‘data‘] = imgData     #将数据写入文件的主键data下面
f[‘labels‘] = np.array([1,2,3,4,5])   #将数据写入文件的主键labels下面
f.close()       #关闭文件

但是，这种方法不适用我的问题，我想知道如何存储 类似字典中的字典 的数据形式。test ：x:test_x,  y:test_y       train : x : train_x, y: train_y

解决方法：

import h5py
import numpy as np
#HDF5的写入：
imgData = np.zeros((2,4))
f = h5py.File(‘HDF5_FILE.h5‘,‘w‘) #创建一个h5文件，文件指针是f

grap1=f.create_group("test")
grap2=f.create_group(‘train‘)

grap1.create_dataset("x",data=test_x)
grap1.create_dataset("y",data=test_y)

grap2.create_dataset("x",data=train_x)
grap2.create_dataset("y",data=train_y)

f.close()       #关闭文件