温馨提示×

python eye函数的语法结构解析

小樊
84
2024-09-08 21:58:23
栏目: 编程语言

numpy.eye 是 NumPy 库中的一个函数,用于创建一个对角线上元素为 1,其余元素为 0 的二维数组(也称为单位矩阵)

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

参数说明:

  • N:输出矩阵的行数。必需参数。
  • M:输出矩阵的列数。如果未指定,默认值为 N,即创建一个方阵。
  • k:指定对角线的位置。正值表示上对角线,负值表示下对角线,0 表示主对角线。默认值为 0。
  • dtype:输出矩阵的数据类型。默认值为 float
  • order:输出矩阵的内存布局。可以是 'C'(按行优先存储)或 'F'(按列优先存储)。默认值为 'C'
  • like:用于指定输出矩阵的类型和属性。如果提供了此参数,将根据 like 的类型和属性创建输出矩阵。

示例:

import numpy as np

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

# 创建一个 4x4 的单位矩阵,其中对角线上的元素为 2
scaled_identity_matrix = np.eye(4, dtype=int) * 2
print(scaled_identity_matrix)

# 创建一个 4x4 的矩阵,其中第 2 条对角线上的元素为 1
diagonal_matrix = np.eye(4, k=1)
print(diagonal_matrix)

输出:

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

[[2 0 0 0]
 [0 2 0 0]
 [0 0 2 0]
 [0 0 0 2]]

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

0