如何在python中使用eye函数

avatar
作者
筋斗云
阅读量:0

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

广告一刻

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