如何使用h5py讀入數據

在網上找到的方法，以下：

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() #關閉文件