numpy.ma.identity

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

返回单位数组.

单位数组是一个主对角线上为 1 的方阵.

参数:

nint

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

dtypedata-type,可选

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

like类数组,可选

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

在 1.20.0 版本加入.

返回:

outMaskedArray

一个 n x n 的数组,其主对角线设置为 1,所有其他元素为 0.

示例

Try it in your browser!

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

Go BackOpen In Tab