NumPy 函数

NumPy matlib.eye() 函数返回一个矩阵，对角线上有 1，其他位置有 0。

语法

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

    参数

    n必填。 指定矩阵中的行数。
    M可选。 指定矩阵中的列数。默认值为 n。
    k可选。 指定对角线的索引。 0 表示主对角线，正值表示上对角线，负值表示下对角线。
    dtype可选。 指定矩阵所需的数据类型。默认值:浮动
    order可选。 指定是否存储结果。两个可能的值是:C（C 风格）和 F（Fortran 风格）。默认值:'C'

    返回值

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

    示例:

    在下面的示例中，matlib.eye()函数用于使用不同的可用参数创建矩阵。

    import numpy as np
import numpy.matlib

#使用默认参数
mat1 = np.matlib.eye(n=3)
print("mat1 is:\n", mat1)

#使用正 k
mat2 = np.matlib.eye(n=3, M=4, k=1)
print("\nmat2 is:\n", mat2)

#使用负 k 和 int 数据类型
mat3 = np.matlib.eye(n=3, M=4, k=-1, dtype=int)
print("\nmat3 is:\n", mat3) 
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13

    输出上面的代码将是:

    mat1 is:
 [[1. 0. 0.]
  [0. 1. 0.]
  [0. 0. 1.]]

mat2 is:
 [[0. 1. 0. 0.]
  [0. 0. 1. 0.]
  [0. 0. 0. 1.]]

mat3 is:
 [[0 0 0 0]
  [1 0 0 0]
  [0 1 0 0]] 
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13