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.]]
来源：CSDN
作者：zhenyu wu
链接：https://blog.csdn.net/wzy628810/article/details/103814725