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