📅  最后修改于: 2020-06-01 12:13:00             🧑  作者: Mango
numpy.eye(R,C =无,K = 0,D型细胞类型= < ‘浮动’>):返回具有在对角线上的1和0的别处WRT矩阵ķ。
参数:
R:行数
C:[可选]列数;默认情况下M = N
k:[int,可选,默认为0]
我们需要对角线; k> 0表示对角线高于主对角线,反之亦然。
dtype:[可选,float(默认)]返回数组的数据类型。
返回值:
形状数组R x C,其中所有元素 均等于0 的数组,第k个对角线除外, 其值等于1。
# Python编程说明numpy.eye方法
import numpy as geek
# 2x2矩阵,主要诊断为1
b = geek.eye(2, dtype = float)
print("Matrix b : \n", b)
# 在主要诊断以下的诊断中,R = 4,C = 5且矩阵为1的矩阵
a = geek.eye(4, 5, k = -1)
print("\nMatrix a : \n", a)
输出:
矩阵b:
[[1. 0.]
[0. 1.]]
矩阵a:
[[0. 0. 0. 0. 0.]
[1. 0. 0. 0. 0.]
[0. 1。 0. 0. 0.]
[0. 0. 1. 0. 0.]]