python中利用h5py模块读取h5文件中的主键方法

如下所示:

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()       #关闭文件
#HDF5的读取:
f = h5py.File('HDF5_FILE.h5','r') #打开h5文件
# 可以查看所有的主键
for key in f.keys():
 print(f[key].name)
 print(f[key].shape)
 print(f[key].value)

输出结果:

E:\phthon35\python.exe I:/catsVSdogs1-master/catsVSdogs1-master/file01/test10.py
/data
(2, 4)
[[ 0. 0. 0. 0.]
 [ 0. 0. 0. 0.]]
/labels
(5,)
[1 2 3 4 5]
Process finished with exit code 0