SciPy Matlab 数组
使用 Matlab 数组
我们知道 NumPy 为我们提供了将数据保存为 Python 可读格式的方法。 但 SciPy 也为我们提供了与 Matlab 的互操作性。
SciPy 为我们提供了模块 scipy.io，它具有处理 Matlab 数组的功能。
以 Matlab 格式导出数据
savemat()函数允许我们以Matlab格式导出数据。
该方法采用以下参数:
- 文件名 - 保存数据的文件名。
- mdict - 包含数据的字典。
- do_compression - 一个布尔值，指定是否压缩结果。 默认为 False。
实例
将以下数组作为变量名"vec"导出到mat文件:
from scipy import io
import numpy as np
arr = np.arange(10)
io.savemat('arr.mat', {"vec": arr})
注释:上面的示例在您的计算机上保存了一个文件名"arr.mat"。
要打开文件，请查看下面的"从 Matlab 格式导入数据"示例:
从 Matlab 格式导入数据
loadmat() 函数允许我们从 Matlab 文件中导入数据。
该函数需要一个必需参数:
filename - 保存数据的文件名。
它会返回一个结构化数组，其键是变量名，对应的值是变量值。
实例
从以下 mat 文件中导入数组:
from scipy import io
import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9,])
# Export:
io.savemat('arr.mat', {"vec": arr})
# Import:
mydata = io.loadmat('arr.mat')
print(mydata)
结果:
{
'__header__': b'MATLAB 5.0 MAT-file Platform: nt, Created on: Tue Sep 22 13:12:32 2020',
'__version__': '1.0',
'__globals__': [],
'vec': array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
}
亲自试一试 »
使用变量名"vec"只显示来自matlab数据的数组:
注释:我们可以看到数组最初是一维的，但在提取时它增加了一维。
为了解决这个问题，我们可以传递一个额外的参数 squeeze_me=True:
实例
# Import:
mydata = io.loadmat('arr.mat', squeeze_me=True)
print(mydata['vec'])
结果:
[0 1 2 3 4 5 6 7 8 9]亲自试一试 »
