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

Numpy 的 eye(~) 方法返回一个 2D 数组，其对角线填充 1，所有其他条目填充 0。

1. N | int

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

2. M | int | optional

所需的列数。默认情况下，M=N。

3. k | int | optional

对角线的偏移量。如果为正，则对角线将向上移动，否则对角线将向下移动。默认情况下，k=0 。

4. dtype | string 或 type | optional

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

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

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

np.eye(3)



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

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

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



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

要创建具有 4 列的数组，请按如下方式设置 M=4：

np.eye(3, M=4)



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

要将对角线向上移动 1：

np.eye(3, k=1)



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

要将对角线向下移动 2：

np.eye(3, k=-2)



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

相关用法

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