Jupyter Notebook 中如何导入和使用 H5 文件数据
Jupyter Notebook 中如何导入和使用 H5 文件数据
在 Jupyter Notebook 中导入 H5 格式数据,需要使用 h5py 库。可以使用以下代码进行导入:
import h5py
# 打开 h5 文件
file = h5py.File('filename.h5', 'r')
# 查看文件中的数据集
for key in file.keys():
print(key)
# 获取数据集
dataset = file['dataset_name']
# 将数据集转换为 NumPy 数组
data = dataset[()]
# 关闭文件
file.close()
在上面的代码中,'filename.h5' 是 h5 文件的名称,'dataset_name' 是要获取的数据集的名称。data 是一个 NumPy 数组,其中包含数据集的所有值。
如果需要将数据集写入 H5 文件中,可以使用 h5py.File 的 create_dataset 方法。例如:
import h5py
import numpy as np
# 创建 h5 文件
file = h5py.File('newfile.h5', 'w')
# 创建数据集
data = np.array([1, 2, 3])
dataset = file.create_dataset('dataset_name', data=data)
# 关闭文件
file.close()
在上面的代码中,'newfile.h5' 是新文件的名称,'dataset_name' 是要创建的数据集的名称,data 是要写入数据集的值。
原文地址: https://www.cveoy.top/t/topic/jNsF 著作权归作者所有。请勿转载和采集!