NumPy 函数

NumPy identity() 函数返回恒等数组。恒等数组是一个方阵,主对角线上有 1。使用该函数的语法如下:

语法

numpy.identity(n, dtype=None) 

    参数

    n必填。 指定 n x n 输出中的行数(和列数)。
    dtype可选。 指定输出所需的数据类型。默认值:float。

    返回值

    返回 n x n 数组,其主对角线设置为 1,所有其他元素设置为 0。

    示例:

    下面的示例显示了identity()函数的用法。

    import numpy as np
    
    Arr1 = np.identity(3)
    print("Arr1 is:\n", Arr1)
    
    #指定int数据类型
    Arr2 = np.identity(3, dtype=int)
    print("\nArr2 is:\n", Arr2) 
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7

    上述代码的输出将是:

    Arr1 is:
    [[1. 0. 0.]
     [0. 1. 0.]
     [0. 0. 1.]]
    
    Arr2 is:
    [[1 0 0]
     [0 1 0]
     [0 0 1]] 
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8