numpy.ma.identity#
- ma.identity(n, dtype=None) = <numpy.ma.core._convert2ma object>#
返回单位数组。
单位数组是一个对角线上为一,其他元素为零的方阵。
- 参数:
- nint
输出 n x n 数组的行数(和列数)。
- dtype数据类型, 可选
输出的数据类型。默认为
float
。- like类数组对象, 可选
参考对象,允许创建非 NumPy 数组的数组。如果作为
like
传入的类数组对象支持__array_function__
协议,则结果将由它定义。在这种情况下,它确保创建与通过此参数传入的对象兼容的数组对象。1.20.0 版本新增。
- 返回:
- out掩码数组
一个 n x n 数组,其主对角线元素为一,所有其他元素为零。
示例
>>> import numpy as np >>> np.identity(3) array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])