numpy.identity#

numpy.identity(n, dtype=None, *, like=None)[source]#

返回单位阵。

单位阵是一个方形数组,主对角线上的元素为 1。

参数:
nint

n x n 输出中的行数(和列数)。

dtype数据类型,可选

输出的数据类型。默认为 float

like类数组,可选

参考对象,允许创建非 NumPy 数组的数组。如果作为 like 传入的类数组支持 __array_function__ 协议,则结果将由其定义。在这种情况下,它确保创建与通过此参数传入的数组对象兼容的数组对象。

1.20.0 版中的新内容。

返回值:
outndarray

n x n 数组,其主对角线设置为 1,所有其他元素为 0。

示例

>>> import numpy as np
>>> np.identity(3)
array([[1.,  0.,  0.],
       [0.,  1.,  0.],
       [0.,  0.,  1.]])