1、创建引入库并创建h5文件
import h5py
import numpy as np
file_name='data.h5'
h5f=h5py.File(file_name)
2、批量写入数据的方法(支持任意维度的数据)一直追加数据到h5文件中
def save_h5(h5f,data,target):
shape_list=list(data.shape)
if not h5f.__contains__(target):
shape_list[0]=None #设置数组的第一个维度是0
dataset = h5f.create_dataset(target, data=data,maxshape=tuple(shape_list), chunks=True)
return
else:
dataset = h5f[target]
len_old=dataset.shape[0]
len_new=len_old+data.shape[0]
shape_list[0]=len_new
dataset.resize(tuple(shape_list)) #修改数组的第一个维度
dataset[len_old:len_new] = data #存入新的文件
3、调用批量写入的方法 (注意da