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, 可选

对角线放置位置。默认值：0（主对角线）。

dtypedtype, 可选

数组的类型

格式str, 可选 (默认: “dia”)

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

示例

Try it in your browser!

>>> 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)>

Go BackOpen In Tab