python3 读取numpy和写入numpy到hdf5

在用深度学习训练的时候，我们经常要保存一些数据，用pickle的话，最大只能保存４GB，但是hdf5是没有限制的，所以这里介绍一下如果用python的h5py库来读取数据和写入数据。

首先要安装h5py:

pip install h5py

写入

import numpy as np
import h5py

data_to_write = np.random.random(size=(100,100)) 

with h5py.File('name-of-file.h5', 'w') as hf:
    hf.create_dataset("name-of-dataset",  data=data_to_write)

读取

with h5py.File('name-of-file.h5', 'r') as hf:
    data = hf['name-of-dataset'][:]

参考文献

[1].Input and output numpy arrays to h5py. ​​https://stackoverflow.com/questions/20928136/input-and-output-numpy-arrays-to-h5py​​