在网上找到的方法,如下:
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() #关闭文件