当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python numpy eye用法及代码示例


本文简要介绍 python 语言中 numpy.eye 的用法。

用法:

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

返回一个二维数组,其中对角线为 1,其他位置为 0。

参数

N int

输出中的行数。

M 整数,可选

输出中的列数。如果没有,则默认为 N。

k 整数,可选

对角线索引:0(默认)表示主对角线,正值表示上对角线,负值表示下对角线。

dtype 数据类型,可选

返回数组的数据类型。

order {‘C’, ‘F’},可选

输出是否应按行优先(C 风格)或列优先(Fortran-style)顺序存储在内存中。

like array_like

允许创建非 NumPy 数组的引用对象。如果作为 like 传入的类似数组支持 __array_function__ 协议,则结果将由它定义。在这种情况下,它确保创建一个与通过此参数传入的数组对象兼容的数组对象。

返回

I 形状的ndarray(N,M)

一个数组,其中所有元素都等于 0,除了 k-th 对角线,其值等于 1。

例子

>>> np.eye(2, dtype=int)
array([[1, 0],
       [0, 1]])
>>> np.eye(3, k=1)
array([[0.,  1.,  0.],
       [0.,  0.,  1.],
       [0.,  0.,  0.]])

相关用法


注:本文由纯净天空筛选整理自numpy.org大神的英文原创作品 numpy.eye。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。