Python NumPy identity方法用法及代码示例

Numpy 的 identity(~) 方法返回指定大小的单位矩阵。

1. n | int

矩阵的大小(即行数和列数)。

2. dtype | string 或 type | optional

返回的单位矩阵所需的数据类型。默认情况下，dtype=Float 。

表示单位矩阵的 2D Numpy 数组。

创建大小为 3 的单位矩阵：

np.identity(3)



array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

要创建 int64 类型的单位矩阵：

np.identity(3, dtype="int64")



array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | identity method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。