scipy.sparse.
identity#
- scipy.sparse.identity(n, dtype='d', format=None)[源代码]#
稀疏格式的单位矩阵
使用给定的稀疏格式和 dtype 返回形状为 (n,n) 的单位矩阵。这与
eye_array
的不同之处在于它具有正方形形状,并且只在主对角线上有 1。因此它是乘法单位矩阵。eye_array
允许矩形形状,并且对角线可以从主对角线偏移。警告
此函数返回一个稀疏矩阵 – 而不是稀疏数组。建议您使用
eye_array
来利用稀疏数组的功能。- 参数:
- nint
单位矩阵的形状。
- dtypedtype,可选
矩阵的数据类型
- formatstr,可选
结果的稀疏格式,例如,format=”csr” 等。
示例
>>> import scipy as sp >>> sp.sparse.identity(3).toarray() array([[ 1., 0., 0.], [ 0., 1., 0.], [ 0., 0., 1.]]) >>> sp.sparse.identity(3, dtype='int8', format='dia') <DIAgonal sparse matrix of dtype 'int8' with 3 stored elements (1 diagonals) and shape (3, 3)> >>> sp.sparse.eye_array(3, dtype='int8', format='dia') <DIAgonal sparse array of dtype 'int8' with 3 stored elements (1 diagonals) and shape (3, 3)>