scipy.sparse.

eye#

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

對角線為 1 的稀疏矩陣

傳回一個稀疏矩陣 (m x n)，其中第 k 條對角線全為 1，其他地方全為零。

參數:

mint: 矩陣中的列數。
nint, optional: 行數。預設值：m。
kint, optional: 要放置 1 的對角線。預設值：0（主對角線）。
dtypedtype, optional: 矩陣的資料類型。
formatstr, optional: 結果的稀疏格式，例如 format=”csr” 等。
.. warning:: 此函數傳回稀疏矩陣，而非稀疏陣列。建議您使用 eye_array 以利用稀疏陣列功能。

範例

>>> import numpy as np
>>> import scipy as sp
>>> sp.sparse.eye(3).toarray()
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
>>> sp.sparse.eye(3, dtype=np.int8)
<DIAgonal sparse matrix of dtype 'int8'
    with 3 stored elements (1 diagonals) and shape (3, 3)>