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