h5py将MATLAB中mat结构体转换为Python的dict形式

最新推荐文章于 2022-11-26 22:17:17 发布

xtphysics

最新推荐文章于 2022-11-26 22:17:17 发布

阅读量583

点赞数

分类专栏：数据读写文章标签： python matlab 字符串

本文链接：https://blog.csdn.net/kouwang9779/article/details/118543838

版权

数据读写专栏收录该内容

5 篇文章 0 订阅

订阅专栏

这段代码展示了如何利用h5py库将MATLAB的.mat文件中的结构体转换成Python的字典。通过读取MATLAB文件，遍历结构体的键，根据值的类型（数值或字符串）将其转换为numpy数组并存储到字典中。

摘要由CSDN通过智能技术生成

h5py将MATLAB中mat结构体转换为Python的dict形式，Python代码如下：

def mat2dict(path, strname):
    struct_mat = h5py.File(path,'r')# 将MATLAB结构体读入
    struct_dict = dict()
    struct_keys = list(struct_mat[strname].keys()) # 结构体中所有的key
    for i in range(len(struct_keys)):
        value_list = []
        if strname is not None:
            # 如果字典中value不是数值（是字符串的情况）
            if struct_mat[strname][struct_keys[i]].dtype != '<f8':
                strlist = [u''.join(chr(c) for c in struct_mat[strname][struct_keys[i]])]
                value_list.append(strlist)
                temp_dict = {
                    struct_keys[i]:np.array(value_list)
                }
                struct_dict.update(temp_dict)
            else: # 如果字典中value是数值的情况
                num = len(struct_mat[strname][struct_keys[i]][()]) # 可能一个key对应的value中有多个值
                for j in range(num):
                    value_list.append(struct_mat[strname][struct_keys[i]][()][j][0])
                    temp_dict = {
                        struct_keys[i]:np.array(value_list)
                    }
                    struct_dict.update(temp_dict)
    return struct_dict