scipy.sparse.
eye#
- scipy.sparse.eye(m, n=None, k=0, dtype=<class 'float'>, format=None)[source]#
对角线元素为 1 的稀疏矩阵
返回稀疏矩阵 (m x n),其中第 k 条对角线的所有元素均为 1,其余元素均为 0。
- 参数:
- mint
矩阵的行数。
- nint,可选
列数。默认值:m。
- kint,可选
指定在对角线元素为 1 的位置。默认值:0(主对角线)。
- dtypedtype,可选
矩阵的数据类型。
- formatstr,可选
稀疏格式,例如格式=”csr”等。
- .. 警告:
此函数返回稀疏矩阵,而不是稀疏数组。建议使用
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)>