将加载的MAT文件转换回NumPy数组 [英] Convert loaded mat file back to numpy array

问题描述

我使用scipy.io.savemat()将大小为5000，96，96的NumPy数组中的图像保存到.mat文件中。

当我想要将这些图像加载回Python时，我使用scipy.io.loadmat()，但这次它们被放入词典中。

如何才能将它们从字典整齐地放到NumPy数组中？

我正在使用scipy.io.loadmat加载MATLAB文件，并希望将其放入NumPy数组中。图像的亮度为(5000，96，96)

scipy.io.savemat("images.mat")
z = scipy.io.loadmat("images.mat")

NumPy数组中的图像

推荐答案

保存3D数组：

In [53]: from scipy import io                                                   
In [54]: arr = np.arange(8*3*3).reshape(8,3,3)                                  
In [56]: io.savemat('threed.mat',{"a":arr})                                     

加载：

In [57]: dat = io.loadmat('threed.mat')                                         
In [58]: list(dat.keys())                                                       
Out[58]: ['__header__', '__version__', '__globals__', 'a']

按键访问数组(正常字典操作)：

In [59]: dat['a'].shape                                                         
Out[59]: (8, 3, 3)
In [61]: np.allclose(arr,dat['a'])                                              
Out[61]: True

这篇关于将加载的MAT文件转换回NumPy数组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！