import os
import numpy as np
import hdf5
TRAIN_FEATURE_DIR = "MSVD"
for filename in os.listdir(TRAIN_FEATURE_DIR):
f = np.load(os.path.join(TRAIN_FEATURE_DIR, filename))
...
import glob
import numpy as np
import h5py
with h5py.File('SO_74788877.h5','w') as h5f:
for filename in glob.iglob('*.npy'):
arr = np.load(filename)
h5f.create_dataset(filename,data=arr)
2条答案
按热度按时间bwntbbo31#
从数组创建数据集很简单。下面的示例循环遍历文件夹中的所有
.npy
文件,并为每个数组创建一个数据集。(仅供参考,我更喜欢glob.iglob()
使用通配符获取文件名。)数据集名称与文件名相同。以下代码显示了如何从上面创建的H5文件中访问数据集名称和值。(
dataset
是一个数据集对象,在许多情况下,其行为类似于numpy数组):ao218c7q2#
下面的代码解决了我的问题: