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