📜  Python中的numpy.identity

📅  最后修改于: 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.]]