numpy.ma.identity#

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

返回标识数组.

身份数组是一个主对角线上为1的方形数组.

参数:
nint

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

dtype数据类型,可选

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

likearray_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.]])