numpy.identity#

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

返回单位数组。

单位数组是一个方阵，其主对角线元素为一。

参数：

nint: n x n 输出中的行数（和列数）。
dtype数据类型, 可选: 输出的数据类型。默认为 float。
like类数组对象, 可选: 参考对象，允许创建非 NumPy 数组的数组。如果作为 like 传入的类数组对象支持 __array_function__ 协议，则结果将由其定义。在这种情况下，它确保创建与通过此参数传入的对象兼容的数组对象。

1.20.0 版新增。

返回：

示例

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