python 保存成h5文件

在 Python 中，可以使用 h5py 库来将数据保存为 HDF5 格式的文件，其后缀名为 .h5。下面是一个保存 NumPy 数组为 HDF5 文件的示例代码：

import h5py
import numpy as np

# 创建一个 NumPy 数组
data = np.random.rand(100, 100)

# 创建 HDF5 文件并写入数据
with h5py.File('data.h5', 'w') as f:
    f.create_dataset('data', data=data)

在上述代码中，我们首先使用 NumPy 库创建了一个大小为 100x100 的随机数组，然后使用 h5py 库创建了一个名为 'data.h5' 的 HDF5 文件，并将 NumPy 数组写入其中。其中，'data' 是 HDF5 文件中的数据集名称，而 data 参数则是要写入的数据。

如果要将多个 NumPy 数组写入同一个 HDF5 文件中，可以使用如下代码：

import h5py
import numpy as np

# 创建两个 NumPy 数组
data1 = np.random.rand(100, 100)
data2 = np.random.rand(50, 50)

# 创建 HDF5 文件并写入数据
with h5py.File('data.h5', 'w') as f:
    f.create_dataset('data1', data=data1)
    f.create_dataset('data2', data=data2)

在上述代码中，我们创建了两个不同大小的 NumPy 数组，并使用相应的数据集名称将它们保存到同一个 HDF5 文件中。