📅  最后修改于: 2020-06-01 12:12:39             🧑  作者: Mango
numpy.identity(n,dtype = None):返回一个单位矩阵,即在主色块上带有1的方矩阵。
参数:
n:[int]输出数组的尺寸n x n
dtype:[可选,float(默认)]返回数组的数据类型。
返回值:
# Python编程说明numpy.identity方法
import numpy as geek
# 2x2矩阵,主要诊断为1
b = geek.identity(2, dtype = float)
print("Matrix b : \n", b)
a = geek.identity(4)
print("\nMatrix a : \n", a)
输出:
矩阵b:
[[1. 0.]
[0. 1.]]
矩阵a:
[[1. 0. 0. 0.]
[0. 1. 0. 0.]
[0. 0. 1. 0。 ]
[0. 0. 0. 1.]]