Python SciPy sparse.eye用法及代碼示例

本文簡要介紹 python 語言中 scipy.sparse.eye 的用法。

用法:

scipy.sparse.eye(m, n=None, k=0, dtype=<class 'float'>, format=None)#

對角線上的稀疏矩陣

返回一個稀疏 (m x n) 矩陣，其中第 k 個對角線全為 1，其他全為 0。

參數 ：：

m： int

矩陣中的行數。

n： 整數，可選

列數。默認值：米。

k： 整數，可選

放置的對角線。默認值：0(主對角線)。

dtype： dtype，可選

矩陣的數據類型。

format： str，可選

結果的稀疏格式，例如format=”csr”等。

例子：

>>> import numpy as np
>>> from scipy import sparse
>>> sparse.eye(3).toarray()
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
>>> sparse.eye(3, dtype=np.int8)
<3x3 sparse matrix of type '<class 'numpy.int8'>'
    with 3 stored elements (1 diagonals) in DIAgonal format>