python

python eye函数与其他矩阵函数的比较

小樊
82
2024-09-08 21:54:55
栏目: 编程语言

eye 函数是 NumPy 库中用于创建单位矩阵(identity matrix)的一个函数

以下是 eye 函数与其他矩阵函数的比较:

  1. numpy.eye(N, M=None, k=0, dtype=float):创建一个 NxM 的单位矩阵,其中 N 是行数,M 是列数(默认为 N),k 是对角线索引(0 表示主对角线,正数表示上对角线,负数表示下对角线)。dtype 参数指定矩阵元素的数据类型。
import numpy as np
I = np.eye(3)
print(I)

输出:

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
  1. numpy.zeros((N, M), dtype=float):创建一个 NxM 的全零矩阵。
Z = np.zeros((3, 3))
print(Z)

输出:

[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
  1. numpy.ones((N, M), dtype=float):创建一个 NxM 的全一矩阵。
O = np.ones((3, 3))
print(O)

输出:

[[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]
  1. numpy.diag(v, k=0):从给定向量 v 创建一个对角矩阵,k 参数表示对角线索引。
D = np.diag([1, 2, 3])
print(D)

输出:

[[1 0 0]
 [0 2 0]
 [0 0 3]]

总结:eye 函数专门用于创建单位矩阵,而其他矩阵函数可以创建全零、全一或自定义对角线的矩阵。在需要单位矩阵时,使用 eye 函数更直观和高效。

0
看了该问题的人还看了