Python中numpy.identity()函数
numpy.identity(n, dtype = None):返回一个单位矩阵i.e。主对角线上为1的方阵。
Parameters :
n : [int] n * n维的输出数组
dtype : [可选, float(by Default)] 返回数组的数据类型。
返回值:
n x n的单位数组,其主对角线设置为1,其他所有元素为0。
示例
# Python Programming illustrating
# numpy.identity method
import numpy as geek
# 2x2 matrix with 1's on main diagonal
b = geek.identity(2, dtype = float)
print("Matrix b : \n", b)
a = geek.identity(4)
print("\nMatrix a : \n", a)
输出 :
Matrix b :
[[ 1. 0.]
[ 0. 1.]]
Matrix a :
[[ 1. 0. 0. 0.]
[ 0. 1. 0. 0.]
[ 0. 0. 1. 0.]
[ 0. 0. 0. 1.]]