温馨提示×

python中eye函数的作用是什么

小樊
115
2024-09-08 21:49:07
栏目: 编程语言

在Python中,eye函数通常指的是NumPy库中的numpy.eye函数

numpy.eye函数用于创建一个对角线上元素为1,其他元素为0的二维数组(也称为单位矩阵)。函数的基本语法如下:

numpy.eye(N, M=None, k=0, dtype=float, order='C', *, like=None)

参数说明:

  • N:输出矩阵的行数。
  • M:输出矩阵的列数。默认值为None,表示列数等于行数。
  • k:指定对角线的索引。正值表示上对角线,负值表示下对角线,0表示主对角线。
  • dtype:输出矩阵的数据类型。默认为float
  • order:输出矩阵的存储顺序。'C'表示按行存储,'F'表示按列存储。
  • like:用于指定输出矩阵的类型。如果提供了like参数,则输出矩阵将与like参数指定的矩阵具有相同的类型。

以下是一个使用numpy.eye函数的示例:

import numpy as np

# 创建一个3x3的单位矩阵
identity_matrix = np.eye(3)
print(identity_matrix)

输出结果:

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

这个示例中,我们创建了一个3x3的单位矩阵,其对角线上的元素为1,其他元素为0。

0