numpy eye 用法及代码示例

用法:

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

返回一个二维数组,对角线上有一个,其他位置为零。

描述
参数 N: : int

输出中的行数。
M: : int, 可选参数

输出中的列数。如果为None,则默认为N。
k: : int, 可选参数

对角线的索引:0(默认值)是指主对角线,正值是指上对角线,负值是指下对角线。
dtype: : data-type, 可选参数

返回数组的数据类型。
order: : {‘C’, ‘F’}, 可选参数

输出应按行优先(C-style)还是列优先(Fortran-style)的顺序存储在内存中。

1.14.0版中的新功能。

返回值: I: : 形状的数组(N,M)

一个数组,其中所有元素均等于零(k-th对角线除外,其值等于一)。

例子:

>>> 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.]])

发表回复

登录... 后才能评论