numpy.eye

numpy.eye(N, M=None, k=0, dtype=<class 'float'>, order='C', *, like=None)[源代码]

返回一个二维数组,对角线上有1,其他地方有0。

参数
N利息

输出中的行数。

M可选的

输出中的列数。如果没有,则默认为 N .

k可选的

对角线的索引:0(默认)表示主对角线,正值表示上对角线,负值表示下对角线。

dtype数据类型,可选

返回数组的数据类型。

order'C'、'F',可选

输出应该按内存中的行主(C样式)还是列主(Fortran样式)顺序存储。

1.14.0 新版功能.

likearray_like

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

注解

这个 like 关键字是一个实验性的特性,有待接受 NEP 35 .

1.20.0 新版功能.

返回
Id形状排列(N,m)

所有元素都等于零的数组,除了 k -对角线,其值等于1。

参见

identity

(几乎)等效函数

diag

来自用户指定的一维数组的对角二维数组。

实例

>>> np.eye(2, dtype=int)
array([[1, 0],
       [0, 1]])
>>> np.eye(3, k=1)
array([[0.,  1.,  0.],
       [0.,  0.,  1.],
       [0.,  0.,  0.]])