此函数返回一个矩阵,其对角线元素初始化为1,在其他位置初始化为零。
语法
numpy.matlib.eye(n, m, k, dtype)
参数
它接受以下参数。
- n:代表结果矩阵中的行数。
- m:它表示结果矩阵中的数字列。
- k:它是对角线的索引。
- dtype:这是输出的数据类型。
返回
返回具有未初始化条目的矩阵。
例
import numpy as np import numpy.matlib print(numpy.matlib.eye(n=3,m=3,k=0,dtype=int))
输出:
[[1 0 0] [0 1 0] [0 0 1]]
例: 初始化浮点值
import numpy as np import numpy.matlib print(numpy.matlib.eye(n=3,m=3,k=0,dtype=float))
输出:
[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]