Python numpy identity用法及代码示例

本文简要介绍 python 语言中 numpy.identity 的用法。

用法:

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

返回标识数组。

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

参数：

n： int

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

dtype： 数据类型，可选

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

like： array_like

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

返回：

out： ndarray

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

例子：

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