numpy.identity

numpy.identity(n, dtype=None, *, like=None)

返回标识数组。

标识数组是一个正方形数组，在主对角线上有一个。

参数

n利息

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

dtype数据类型，可选

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

likearray_like

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

注解

这个 like 关键字是一个实验性的特性，有待接受 NEP 35 .

1.20.0 新版功能.

返回

out恩达雷

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

实例

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

numpy.eye numpy.ones