scipy.sparse.

eye_array#

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

稀疏数组格式中的单位矩阵

返回对角线处为 1 的稀疏数组。具体来说,是一个稀疏数组(m x n),其中第 k 条对角线全是 1,其他元素全是 0。

参数:
mint 或 int 元组

请求的行数。

nint,可选

列数。默认值:m

kint,可选

放置 1 所在的对角线。默认值:0(主对角线)。

dtypedtype,可选

数组的数据类型

formatstr,可选(默认值:“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)>