NumPy matlib.eye() 函数返回一个矩阵，对角线上有 1，其他位置有 0。
语法
numpy.matlib.eye(n, M=None, k=0, dtype='float', order='C') 参数
| n | 必填。 指定矩阵中的行数。 |
| M | 可选。 指定矩阵中的列数。默认值为 n。 |
| k | 可选。 指定对角线的索引。 0 表示主对角线，正值表示上对角线，负值表示下对角线。 |
| dtype | 可选。 指定矩阵所需的数据类型。默认值:浮动 |
| order | 可选。 指定是否存储结果。两个可能的值是:C（C 风格）和 F（Fortran 风格）。默认值:'C' |
返回值
返回一个矩阵，对角线上有 1，其他位置有 0。
示例:
在下面的示例中，matlib.eye()函数用于使用不同的可用参数创建矩阵。
import numpy as np
import numpy.matlib
#使用默认参数
mat1 = np.matlib.eye(n=3)
print("mat1 is:\n", mat1)
#使用正 k
mat2 = np.matlib.eye(n=3, M=4, k=1)
print("\nmat2 is:\n", mat2)
#使用负 k 和 int 数据类型
mat3 = np.matlib.eye(n=3, M=4, k=-1, dtype=int)
print("\nmat3 is:\n", mat3) 输出上面的代码将是:
mat1 is:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
mat2 is:
[[0. 1. 0. 0.]
[0. 0. 1. 0.]
[0. 0. 0. 1.]]
mat3 is:
[[0 0 0 0]
[1 0 0 0]
[0 1 0 0]]