python中eye函数的作用是什么

avatar
作者
筋斗云
阅读量:0

在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。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!