numpy.eye详解

numpy.eye

• 用例:
  numpy.eye(N, M=None, k=0, dtype=<class ‘float’>, order=‘C’)

• 功能:
  返回一个二维数组，其对角线元素为1，其余位置元素为0。

• 参数

变量名	数据类型	功能
N	整数	返回数组的行数
k	整数，可选参数	对角线的索引：0（默认值）代表主对角线，正整数代表上三角内的对角线，负整数代表下三角内的对角线。
dtype	数值类型，可选参数	返回数组的数值类型
order	{‘C’, 'F}，可选参数	是否在内存中以C或fortran(行或列)顺序存储多维数据，版本1.14.0中的新特性

• 返回值

变量名	数据类型	功能
I	维度为(N,M)的多维数组	除了第k条对角线上元素为1以外，其余元素均为0的数组

• 示例:

import numpy as np
np.eye(2, dtype=int)

---

[[1, 0],
[0, 1]]

---

np.eye(3, k=1)

---

[[0., 1., 0.],
[0., 0., 1.],
[0., 0., 0.]]

---

github链接
https://github.com/wzy6642/numpy-translate