scipy.sparse.

eye_array#

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

稀疏陣列格式的單位矩陣

傳回對角線上為 1 的稀疏陣列。具體來說，是一個 (m x n) 的稀疏陣列，其中第 k 個對角線全為 1，其他地方為零。

參數:

m整數: 請求的列數。
n整數，選填: 欄數。預設值：m。
k整數，選填: 放置 1 的對角線。預設值：0（主對角線）。
dtypedtype，選填: 陣列的資料型別
format字串，選填（預設值：“dia”）: 結果的稀疏格式，例如 format=“csr” 等。

範例

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