温馨提示×

如何在python中使用eye函数

小樊
104
2024-09-08 21:50:08
栏目: 编程语言

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

首先,确保已经安装了 NumPy。如果没有安装,可以使用 pip 进行安装:

pip install numpy

然后,在 Python 代码中使用 eye 函数:

import numpy as np

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

print(identity_matrix)

输出结果将会是:

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

eye 函数还有一些可选参数,例如设置对角线上的值、指定矩阵的形状等。更多关于 eye 函数的信息和用法,请参考 NumPy 官方文档:https://numpy.org/doc/stable/reference/generated/numpy.eye.html

0