from:http://www.aichengxu.com/python/24309.htm
写入:
data = mp.array( [222,333,444] )
label = np.array( [0,1,0] )
img_num = np.array( [0,1,2] )
# 创建HDF5文件
file = h5py.File('TrainSet_rotate.h5','w')
# 写入
file.create_dataset('train_set_x', data = data)
file.create_dataset('train_set_y', data = label)
file.create_dataset('train_set_num',data = img_num)
# 。。。。。。。。。
file.close()
读取:
import numpy as np import h5py # 读方式打开文件 file=h5py.File('TrainSet_rotate.h5','r') # 尽管后面有 '[:]', 但是矩阵怎么进去的就是怎么出来的,不会被拉长(matlab后遗症) train_set_data = file['train_set_x'][:] train_set_y = file['train_set_y'][:] train_set_img_num = file['train_set_img_num'][:] # ......... file.close()