我用h5py做了一个数据集：

f = h5py.File("experimentReadings.hdf5", "w")
dset = f.create_dataset("physics", (5,4), dtype='f')

我有一个变量名列表：namesList = ['height', 'mass', 'velocity', 'gravity']。

我希望这些变量名成为dset中的列的名称。

目前，列的名称只有数字0,1,2,3，如下所示：

我想要这个：

我想我要查找的代码如下：

dset[:,0].column_name = namesList[0]
dset[:,1].column_name = namesList[1]
etc...

无论解决方案是什么，它都需要处理我正在使用的实际数据集，其中namesList是28万字长。

推荐答案

关于数据集大小的有趣问题。我见过10e6行的HDF5文件，但没有看到28万列。您必须进行测试。

关于列/文件的名称，您可以使用记录数组(如hpaulj所述)。使用NumPy数据类型定义名称。我创建了一些任意数据来填充我的rec数组，然后用data=参数引用。

试一试：

# Create some data
data1  = np.arange(100.)
data2  = 2.0*data1
data3  = 3.0*data1
data4  = 3.0*data1

# use namesList to define dtype for recarray
namesList = ['height', 'mass', 'velocity', 'gravity']
ds_dt = np.dtype({'names':namesList,'formats':[(float)]*4 }) 

rec_arr = np.rec.fromarrays([data1, data2, data3, data4], dtype=ds_dt)

with h5py.File("experimentReadings.hdf5", "w") as h5f :

    dset = h5f.create_dataset("physics", (100,), data=rec_arr)