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)

上述代码的输出将是:

Arr1 is:
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

Arr2 is:
[[1 0 0]
 [0 1 0]
 [0 0 1]]