NumPy eye() 函数返回一个二维数组，对角线上有 1，其他位置有 0。
语法
numpy.eye(N, M=None, k=0, dtype='float', order='C') 参数
| N | 必填。 指定输出中的行数。 |
| M | 可选。 指定输出中的列数。默认值为 N。 |
| k | 可选。 指定对角线的索引。 0 表示主对角线，正值表示上对角线，负值表示下对角线。 |
| dtype | 可选。 指定输出所需的数据类型。默认值:浮动。 |
| order | 可选。 指定是否存储结果。两个可能的值是:C（C 风格）和 F（Fortran 风格）。默认值:'C' |
返回值
返回一个二维数组，对角线上有 1，其他位置有 0。
示例:
在下面的示例中，eye() 函数用于使用不同的可用参数创建二维数组。
import numpy as np
#使用默认参数
Arr1 = np.eye(N=3)
print("Arr1 is:\n", Arr1)
#使用正 k
Arr2 = np.eye(N=3, M=4, k=1)
print("\nArr2 is:\n", Arr2)
#使用负 k 和 int 数据类型
Arr3 = np.eye(N=3, M=4, k=-1, dtype=int)
print("\nArr3 is:\n", Arr3) 上述代码的输出将是:
Arr1 is:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
Arr2 is:
[[0. 1. 0. 0.]
[0. 0. 1. 0.]
[0. 0. 0. 1.]]
Arr3 is:
[[0 0 0 0]
[1 0 0 0]
[0 1 0 0]]