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

Пользователь

от jensen , в категории: Python , месяц назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от jorge , месяц назад

@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]]