h5py 模块

原创 python

# -*- 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,)
'''

本文由 Yonghui Wang 创作，采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Dec 19, 2024 12:06 pm