numpy.ma.identity#

ma.identity(n, dtype=None, *, like=None, fill_value=None, hardmask=False)[源代码]#

返回单位矩阵。

单位矩阵是一个主对角线为 1 的方阵。

参数:
nint

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

dtype数据类型,可选

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

likearray_like, optional

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

版本 1.20.0 中新增。

返回:
outMaskedArray

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

示例

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