[人工智能]h5py模块

# -*- coding: utf-8 -*-

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)
'''
/data
(2, 4)
/labels
(5,)
'''