numpy.ma.identity #

嘛。身份( n , dtype = None ) = <numpy.ma.core._convert2ma对象> #

返回恒等数组。

恒等数组是一个方阵,主对角线上有 1。

参数
整数

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

dtype数据类型,可选

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

类似array_like,可选

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

1.20.0 版本中的新增功能。

返回
输出屏蔽数组

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

例子

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