SciPy Matlab 数组

---

使用 Matlab 数组

我们知道 NumPy 为我们提供了将数据保存为 Python 可读格式的方法。 但 SciPy 也为我们提供了与 Matlab 的互操作性。

SciPy 为我们提供了模块 scipy.io，它具有处理 Matlab 数组的功能。

---

以 Matlab 格式导出数据

savemat()函数允许我们以Matlab格式导出数据。

该方法采用以下参数:

1. 文件名 - 保存数据的文件名。
2. mdict - 包含数据的字典。
3. 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数据的数组:

实例

...

print(mydata['vec'])

结果:

 [[0 1 2 3 4 5 6 7 8 9]]

亲自试一试 »

注释:我们可以看到数组最初是一维的，但在提取时它增加了一维。

为了解决这个问题，我们可以传递一个额外的参数 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]

亲自试一试 »

---