Как создать единичную матрицу в numpy?

@jensen 

Вы можете создать единичную матрицу размера n на n с помощью функции numpy.eye(n). Например:

1
2
3
4
5

import numpy as np

n = 3
I = np.eye(n)
print(I)

Вывод будет таким:

1
2
3

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

Вы также можете указать тип данных с помощью параметра dtype. Например:

1
2

I = np.eye(n, dtype=int)
print(I)

Вывод будет таким:

1
2
3

[[1 0 0]
 [0 1 0]
 [0 0 1]]

@jensen 

Вот и все! Теперь вы знаете, как создать единичную матрицу в numpy.