Python中的numpy.eye

numpy.eye函数用于创建一个二维数组，其中对角线上的元素为1，其余为0。

用法

numpy.eye函数的用法如下：

numpy.eye(N, M=None, k=0, dtype=<class 'float'>, order='C')

参数说明：

• N：生成数组的行数。
• M：生成数组的列数。默认情况下，M等于N。
• k：对角线相对于主对角线的偏移量。正值表示对角线在主对角线的上方，负值表示对角线在主对角线的下方。默认值为0，表示主对角线。
• dtype：生成数组的数据类型。默认为float。
• order：生成数组的存储顺序。可选值为'C'（按行存储）和'F'（按列存储）。默认值为'C'。

示例

以下示例展示了numpy.eye函数的用法：

import numpy as np

# 创建一个3x3的单位矩阵
a = np.eye(3)
print(a)
# [[1. 0. 0.]
#  [0. 1. 0.]
#  [0. 0. 1.]]

# 创建一个2x4的矩阵，其次对角线为1
b = np.eye(2, 4, k=1)
print(b)
# [[0. 1. 0. 0.]
#  [0. 0. 1. 0.]]

以上示例中，首先我们使用numpy.eye创建了一个3x3的单位矩阵，然后将其打印输出。

其次，我们创建了一个2x4的矩阵，其次对角线为1，然后将其打印输出。

结论

numpy.eye函数是一个非常有用的函数，可以用于创建单位矩阵和其他特殊的矩阵。使用它可以方便地进行矩阵运算和计算。