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