numpy.identity

麻木的。身份( n , dtype = None , * , like = None )

返回恒等数组。

恒等数组是一个方阵，主对角线上有 1。

参数：

整数​

n x n输出中的行数（和列数） 。

dtype数据类型，可选

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

类似array_like，可选

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

1.20.0 版本中的新增功能。

返回：

输出数组

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

例子

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