Python SciPy io.mmread用法及代码示例

本文简要介绍 python 语言中 scipy.io.mmread 的用法。

用法:

scipy.io.mmread(source)#

将矩阵市场 file-like ‘source’ 的内容读取到矩阵中。

参数 ：：

source： str 或 file-like

Matrix Market 文件名(扩展名 .mtx、.mtz.gz)或打开 file-like 对象。

返回 ：：

a： ndarray 或coo_matrix

密集或稀疏矩阵取决于矩阵市场文件中的矩阵格式。

注意：

例子：

>>> from io import StringIO
>>> from scipy.io import mmread

>>> text = '''%%MatrixMarket matrix coordinate real general
...  5 5 7
...  2 3 1.0
...  3 4 2.0
...  3 5 3.0
...  4 1 4.0
...  4 2 5.0
...  4 3 6.0
...  4 4 7.0
... '''

mmread(source) 以 COO 格式的稀疏矩阵形式返回数据。

>>> m = mmread(StringIO(text))
>>> m
<5x5 sparse matrix of type '<class 'numpy.float64'>'
with 7 stored elements in COOrdinate format>
>>> m.A
array([[0., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0.],
       [0., 0., 0., 2., 3.],
       [4., 5., 6., 7., 0.],
       [0., 0., 0., 0., 0.]])

该方法是线程化的。默认线程数等于系统中CPU的数量。使用threadpoolctl覆盖：

>>> import threadpoolctl
>>>
>>> with threadpoolctl.threadpool_limits(limits=2):
...     m = mmread(StringIO(text))