1. 程式人生 > >python讀寫h5檔案

python讀寫h5檔案

原文連結:https://blog.csdn.net/leibaojiangjun1/article/details/53635353 

h5接受的資料是矩陣跟mat方法一致,但是具有更強的壓縮效能
使用hdf5依賴於python的工具包:h5py

import h5py  #匯入工具包
import numpy as np
#HDF5的寫入:
imgData = np.zeros((30,3,128,256))
f = h5py.File('HDF5_FILE.h5','w')   #建立一個h5檔案,檔案指標是f
f['data'] = imgData                 #將資料寫入檔案的主鍵data下面
f['labels'] = range(100)            #將資料寫入檔案的主鍵labels下面
f.close()                           #關閉檔案
 
#HDF5的讀取:
f = h5py.File('HDF5_FILE.h5','r')   #開啟h5檔案
f.keys()                            #可以檢視所有的主鍵
a = f['data'][:]                    #取出主鍵為data的所有的鍵值
f.close()