NumPy matlib.identity() 函数返回给定大小的方单位矩阵。使用该函数的语法如下:
语法
numpy.matlib.identity(n, dtype=None) 参数
| n | 必填。 指定返回单位矩阵的大小。 |
| dtype | 可选。 指定矩阵所需的数据类型。默认值:float |
返回值
返回给定大小的方单位矩阵。
示例:
下面的示例显示了 matlib.identity() 函数的用法。
import numpy as np
import numpy.matlib
mat1 = np.matlib.identity(3)
print("mat1 is:\n", mat1)
#指定int数据类型
mat2 = np.matlib.identity(3, dtype=int)
print("\nmat2 is:\n", mat2) 上述代码的输出将是:
mat1 is:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
mat2 is:
[[1 0 0]
[0 1 0]
[0 0 1]]