Python numpy matlib.eye用法及代码示例

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

用法:

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

返回一个矩阵，其中对角线为 1，其他位置为 0。

参数：

n： int

输出中的行数。

M： 整数，可选

输出中的列数，默认为 n。

k： 整数，可选

对角线索引：0为主对角线，正值表示上对角线，负值表示下对角线。

dtype： dtype，可选

返回矩阵的数据类型。

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

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

返回：

I： 矩阵

一个 n x M 矩阵，其中所有元素都等于 0，除了 k-th 对角线，其值等于 1。

例子：

>>> import numpy.matlib
>>> np.matlib.eye(3, k=1, dtype=float)
matrix([[0.,  1.,  0.],
        [0.,  0.,  1.],
        [0.,  0.,  0.]])