python h5py 读取mat文件

使用 Python 的 h5py 库读取 MAT 文件
为了使用 h5py 库读取 .mat 文件，首先需要安装该库。可以通过 pip 安装：
pip install h5py
登录后复制
导入必要的库
在开始之前，导入所需的库：
import h5py
import numpy as np
登录后复制
打开和关闭 MAT 文件
可以使用上下文管理器 (with 语句) 来安全地打开和关闭文件[^2]。
with h5py.File('example.mat', 'r') as f:
# 在这里处理文件
pass
登录后复制
访问数据集
MATLAB 文件中的数据通常存储为数据集或组。可以直接通过键访问这些数据集[^3]。
with h5py.File('example.mat', 'r') as f:
dataset = f['dataset_name']
data_array = np.array(dataset)
登录后复制
如果不知道具体的数据集名称，可以遍历整个文件结构来查找所有可用的数据集[^4]。
def print_structure(name, obj):
print(f"{name}: {obj}")

with h5py.File('example.mat', 'r') as f:
f.visititems(print_structure)
登录后复制
对于更复杂的 MATLAB 结构体（如元胞数组），可能需要额外的逻辑来解析它们的内容。
with h5py.File('data.mat', 'r') as f:
cell_data = f['cell_name'][:]
登录后复制
获取所有变量名及其维度
要获取所有的变量名以及其对应的尺寸，可以迭代文件对象并检查每个项的形状属性。
with h5py.File('example.mat', 'r') as f:
for key in f.keys():
shape = f[key].shape if hasattr(f[key], "shape") else None
dtype = f[key].dtype
print(f"Variable '{key}' has type {dtype} and shape {shape}")
登录后复制
此方法允许动态了解文件内的内容而无需提前知道具体的变量名称。