numpy.ma.identity

ma.identity(n, dtype=None) = <numpy.ma.core._convert2ma object>

返回单位矩阵。

单位矩阵是一个方阵，其主对角线上的元素为 1。

参数:

nint

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

dtype数据类型，可选

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

like类数组，可选

参考对象，允许创建非 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.]])